Add __repr__ / __str__ to a couple of objects.

Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>

1 files changed, 7 insertions, 0 deletions

diff --git a/tests/test_base.py b/tests/test_base.py
index 97749f6..4f77421 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -56,6 +56,13 @@ class SafetyTestCase(unittest.TestCase):
 
 
 class FactoryTestCase(unittest.TestCase):
+    def testDisplay(self):
+        class TestObjectFactory(base.Factory):
+            FACTORY_FOR = FakeDjangoModel
+
+        self.assertIn('TestObjectFactory', str(TestObjectFactory))
+        self.assertIn('FakeDjangoModel', str(TestObjectFactory))
+
     def testLazyAttributeNonExistentParam(self):
         class TestObjectFactory(base.Factory):
             one = declarations.LazyAttribute(lambda a: a.does_not_exist )