def test_register_without_typename(self): class TestProperty(): pass PropertyType.register(TestProperty, 'TestProperty') self.assertIs(PropertyType('TestProperty'), TestProperty)
def test_register_with_typename(self): class TestProperty(): typename = 'TestProperty' PropertyType.register(TestProperty) self.assertIs(PropertyType('TestProperty'), TestProperty)