Add __repr__ / __str__ to a couple of objects.

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

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/test_containers.py b/tests/test_containers.py
index effb060..6e58573 100644
--- a/tests/test_containers.py
+++ b/tests/test_containers.py
@@ -80,6 +80,15 @@ class LazyStubTestCase(unittest.TestCase):
 
         self.assertRaises(containers.CyclicDefinitionError, getattr, stub, 'one')
 
+    def test_representation(self):
+        class RandomObj(object):
+            pass
+
+        stub = containers.LazyStub({'one': 1, 'two': 2}, target_class=RandomObj)
+        self.assertIn('RandomObj', repr(stub))
+        self.assertIn('RandomObj', str(stub))
+        self.assertIn('one', str(stub))
+
 
 class OrderedDeclarationMock(declarations.OrderedDeclaration):
     pass