def test_bad_extract_class_fields(self): bases = [] attrs = {'i': 5} expected = {'i': IntType()} fields = _extract_fields(bases, attrs) self.assertNotEqual(expected, fields)
def test_extract_subclass_fields(self): class Foo(Model): x = IntType() y = IntType() z = 5 # should be ignored bases = [Foo] attrs = {'i': IntType()} fields = _extract_fields(bases, attrs) expected = { 'i': attrs['i'], 'x': Foo.x, 'y': Foo.y, } self.assertEqual(fields, expected)
def test_extract_class_fields(self): bases = [Model] attrs = {'i': IntType()} fields = _extract_fields(bases, attrs) self.assertEqual(attrs, fields)