refactor is_function

This commit is contained in:
debugtalk
2018-11-15 20:05:57 +08:00
parent a116c651f9
commit bb3dcb2756
3 changed files with 5 additions and 7 deletions

View File

@@ -239,7 +239,7 @@ def load_module_functions(module):
module_functions = {}
for name, item in vars(module).items():
if validator.is_function((name, item)):
if validator.is_function(item):
module_functions[name] = item
return module_functions

View File

@@ -105,10 +105,9 @@ def is_testcase_path(path):
###############################################################################
def is_function(tup):
""" Takes (name, object) tuple, returns True if it is a function.
def is_function(item):
""" Takes item object, returns True if it is a function.
"""
name, item = tup
return isinstance(item, types.FunctionType)

View File

@@ -50,6 +50,5 @@ class TestValidator(unittest.TestCase):
def test_is_function(self):
func = lambda x: x + 1
self.assertTrue(validator.is_function(("func", func)))
self.assertTrue(validator.is_function(("func", validator.is_testcase)))
self.assertTrue(validator.is_function(func))
self.assertTrue(validator.is_function(validator.is_testcase))