1. import unittest

  2. 

  3. from django.utils import inspect

  4. 

  5. 

  6. class Person:

  7.     def no_arguments(self):

  8.         return None

  9. 

  10.     def one_argument(self, something):

  11.         return something

  12. 

  13.     def just_args(self, *args):

  14.         return args

  15. 

  16.     def all_kinds(self, name, address="home", age=25, *args, **kwargs):

  17.         return kwargs

  18. 

  19.     @classmethod

  20.     def cls_all_kinds(cls, name, address="home", age=25, *args, **kwargs):

  21.         return kwargs

  22. 

  23. 

  24. class TestInspectMethods(unittest.TestCase):

  25.     def test_get_callable_parameters(self):

  26.         self.assertIs(

  27.             inspect._get_callable_parameters(Person.no_arguments),

  28.             inspect._get_callable_parameters(Person.no_arguments),

  29.         )

  30.         self.assertIs(

  31.             inspect._get_callable_parameters(Person().no_arguments),

  32.             inspect._get_callable_parameters(Person().no_arguments),

  33.         )

  34. 

  35.     def test_get_func_full_args_no_arguments(self):

  36.         self.assertEqual(inspect.get_func_full_args(Person.no_arguments), [])

  37.         self.assertEqual(inspect.get_func_full_args(Person().no_arguments), [])

  38. 

  39.     def test_get_func_full_args_one_argument(self):

  40.         self.assertEqual(

  41.             inspect.get_func_full_args(Person.one_argument), [("something",)]

  42.         )

  43.         self.assertEqual(

  44.             inspect.get_func_full_args(Person().one_argument),

  45.             [("something",)],

  46.         )

  47. 

  48.     def test_get_func_full_args_all_arguments_method(self):

  49.         arguments = [

  50.             ("name",),

  51.             ("address", "home"),

  52.             ("age", 25),

  53.             ("*args",),

  54.             ("**kwargs",),

  55.         ]

  56.         self.assertEqual(inspect.get_func_full_args(Person.all_kinds), arguments)

  57.         self.assertEqual(inspect.get_func_full_args(Person().all_kinds), arguments)

  58. 

  59.     def test_get_func_full_args_all_arguments_classmethod(self):

  60.         arguments = [

  61.             ("name",),

  62.             ("address", "home"),

  63.             ("age", 25),

  64.             ("*args",),

  65.             ("**kwargs",),

  66.         ]

  67.         self.assertEqual(inspect.get_func_full_args(Person.cls_all_kinds), arguments)

  68.         self.assertEqual(inspect.get_func_full_args(Person().cls_all_kinds), arguments)

  69. 

  70.     def test_func_accepts_var_args_has_var_args(self):

  71.         self.assertIs(inspect.func_accepts_var_args(Person.just_args), True)

  72.         self.assertIs(inspect.func_accepts_var_args(Person().just_args), True)

  73. 

  74.     def test_func_accepts_var_args_no_var_args(self):

  75.         self.assertIs(inspect.func_accepts_var_args(Person.one_argument), False)

  76.         self.assertIs(inspect.func_accepts_var_args(Person().one_argument), False)

  77. 

  78.     def test_method_has_no_args(self):

  79.         self.assertIs(inspect.method_has_no_args(Person.no_arguments), True)

  80.         self.assertIs(inspect.method_has_no_args(Person().no_arguments), True)

  81.         self.assertIs(inspect.method_has_no_args(Person.one_argument), False)

  82.         self.assertIs(inspect.method_has_no_args(Person().one_argument), False)

  83. 

  84.     def test_func_supports_parameter(self):

  85.         self.assertIs(

  86.             inspect.func_supports_parameter(Person.all_kinds, "address"), True

  87.         )

  88.         self.assertIs(

  89.             inspect.func_supports_parameter(Person().all_kinds, "address"),

  90.             True,

  91.         )

  92.         self.assertIs(inspect.func_supports_parameter(Person.all_kinds, "zone"), False)

  93.         self.assertIs(

  94.             inspect.func_supports_parameter(Person().all_kinds, "zone"),

  95.             False,

  96.         )

  97. 

  98.     def test_func_accepts_kwargs(self):

  99.         self.assertIs(inspect.func_accepts_kwargs(Person.just_args), False)

  100.         self.assertIs(inspect.func_accepts_kwargs(Person().just_args), False)

  101.         self.assertIs(inspect.func_accepts_kwargs(Person.all_kinds), True)

  102.         self.assertIs(inspect.func_accepts_kwargs(Person().just_args), False)
About Sign Up Leave Feedback Contact