31 lines
1020 B
Python
31 lines
1020 B
Python
from .samples import my_func, MyClass
|
|
from ...utils.functions import find_class
|
|
from django.test import SimpleTestCase
|
|
|
|
|
|
class FunctionsTest(SimpleTestCase):
|
|
|
|
def test_find_class(self):
|
|
MainType = MyClass
|
|
main_type = MainType(11)
|
|
SubType = MyClass.MySubClass
|
|
sub_type = SubType(13)
|
|
|
|
def local_func(a: int, b: int):
|
|
return a + b
|
|
|
|
# fonction native : None
|
|
self.assertIsNone(find_class(str.join))
|
|
|
|
# fonctions : None
|
|
self.assertIsNone(find_class(local_func))
|
|
self.assertIsNone(find_class(my_func))
|
|
|
|
# méthodes
|
|
self.assertEqual(find_class(MainType.class_met), MainType)
|
|
self.assertEqual(find_class(MainType.static_met), MainType)
|
|
self.assertEqual(find_class(main_type.std_met), MainType)
|
|
self.assertEqual(find_class(SubType.sub_class_met), SubType)
|
|
self.assertEqual(find_class(SubType.sub_static_met), SubType)
|
|
self.assertEqual(find_class(sub_type.sub_std_met), SubType)
|