class TestPrepareItem(unittest.TestCase): """ Test the function which prepares the API documentation result by converting class objects to name-strings. """ def setUp(self): self.apidoc_view = APIDocumentationMaker() def test_custom_class(self): class CustomClass(object): pass assert self.apidoc_view.prepare(CustomClass) == 'CustomClass' def test_dict_obj(self): expected = prep_json({'a': 'object'}) result = self.apidoc_view.prepare({'a': object}) assert result == expected def test_list_obj(self): expected = prep_json(['object']) result = self.apidoc_view.prepare([object]) assert result == expected def test_dict_list(self): expected = prep_json({'a': ['object']}) result = self.apidoc_view.prepare({'a': [object]}) assert result == expected def test_list_dict(self): expected = prep_json([{'a': 'object'}]) result = self.apidoc_view.prepare([{'a': object}]) assert result == expected def test_listof_object(self): assert self.apidoc_view.prepare(ListOf(object)) == 'ListOf(object)' def test_listof_list(self): expected = ( 'ListOf(\n' + prep_json(['object'], 1) + '\n' +')' ) result = self.apidoc_view.prepare(ListOf([object])) assert result == expected def test_listof_dict(self): expected = ( "ListOf(\n" + prep_json({'a': 'object'}, 1) + '\n' + ')' ) result = self.apidoc_view.prepare(ListOf({'a': object})) assert result == expected
class TestPrepareItem(unittest.TestCase): """ Test the function which prepares the API documentation result by converting class objects to name-strings. """ def setUp(self): self.apidoc_view = APIDocumentationMaker() def test_custom_class(self): class CustomClass(object): pass assert self.apidoc_view.prepare(CustomClass) == 'CustomClass' def test_dict_obj(self): expected = prep_json({'a': 'object'}) result = self.apidoc_view.prepare({'a': object}) assert result == expected def test_list_obj(self): expected = prep_json(['object']) result = self.apidoc_view.prepare([object]) assert result == expected def test_dict_list(self): expected = prep_json({'a': ['object']}) result = self.apidoc_view.prepare({'a': [object]}) assert result == expected def test_list_dict(self): expected = prep_json([{'a': 'object'}]) result = self.apidoc_view.prepare([{'a': object}]) assert result == expected def test_listof_object(self): assert self.apidoc_view.prepare(ListOf(object)) == 'ListOf(object)' def test_listof_list(self): expected = ('ListOf(\n' + prep_json(['object'], 1) + '\n' + ')') result = self.apidoc_view.prepare(ListOf([object])) assert result == expected def test_listof_dict(self): expected = ("ListOf(\n" + prep_json({'a': 'object'}, 1) + '\n' + ')') result = self.apidoc_view.prepare(ListOf({'a': object})) assert result == expected