def test_method_data_types(self):
company = Company.objects.create(name="Django")
person = Person.objects.create(first_name="Human",
last_name="User",
company=company)
self.assertEqual(
get_return_data_type(person.get_status_count.__name__), 'Integer')
self.assertEqual(get_return_data_type(person.get_groups_list.__name__),
'List')
def test_method_data_types(self):
"""
We should be able to get a basic idea of the type returned
by a method
"""
company = Company.objects.create(name="Django")
person = Person.objects.create(first_name="Human", last_name="User", company=company)
self.assertEqual(get_return_data_type(person.get_status_count.__name__), 'Integer')
self.assertEqual(get_return_data_type(person.get_groups_list.__name__), 'List')
def _get_methods(model):
# Gather model methods.
methods = []
for func_name, func in model.__dict__.items():
if (inspect.isfunction(func) and len(inspect.getargspec(func)[0]) == 1):
try:
for exclude in MODEL_METHODS_EXCLUDE:
if func_name.startswith(exclude):
raise StopIteration
except StopIteration:
continue
verbose = func.__doc__
# verbose = parse_model_rst(utils.trim_docstring(verbose), 'xraymodel',
# _('model:') + opts.model_name)
methods.append({
'name': func_name,
'data_type': get_return_data_type(func_name),
'verbose': verbose,
})
return methods
def _get_methods(model):
# Gather model methods.
methods = []
for func_name, func in model.__dict__.items():
if (inspect.isfunction(func)
and len(inspect.getargspec(func)[0]) == 1):
try:
for exclude in MODEL_METHODS_EXCLUDE:
if func_name.startswith(exclude):
raise StopIteration
except StopIteration:
continue
verbose = func.__doc__
# verbose = parse_model_rst(utils.trim_docstring(verbose), 'xraymodel',
# _('model:') + opts.model_name)
methods.append({
'name': func_name,
'data_type': get_return_data_type(func_name),
'verbose': verbose,
})
return methods
def test_method_data_types(self):
company = Company.objects.create(name="Django")
person = Person.objects.create(first_name="Human", last_name="User", company=company)
self.assertEqual(get_return_data_type(person.get_status_count.__name__), 'Integer')
self.assertEqual(get_return_data_type(person.get_groups_list.__name__), 'List')
