Merge pull request faif#131 from fkromer/test_flyweight

faif · faif · commit 9ccf7f1b5f57 · 2016-03-08T20:52:50.000+01:00
tests for flyweight.py
diff --git a/test_flyweight.py b/test_flyweight.py
@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from flyweight import Card
+from sys import version_info
+
+if version_info < (2, 7):
+    import unittest2 as unittest
+else:
+    import unittest
+
+class TestCard(unittest.TestCase):
+
+    def test_instances_shall_reference_same_object(self):
+        c1 = Card('9', 'h')
+        c2 = Card('9', 'h')
+        self.assertEqual(c1, c2)
+        self.assertEqual(id(c1), id(c2))
+
+    def test_instances_with_different_suit_shall_reference_different_objects(self):
+        c1 = Card('9', 'a')
+        c2 = Card('9', 'b')
+        self.assertNotEqual(id(c1), id(c2))
+
+    def test_instances_with_different_values_shall_reference_different_objects(self):
+        c1 = Card('9', 'h')
+        c2 = Card('A', 'h')
+        self.assertNotEqual(id(c1), id(c2))
+
+    def test_instances_shall_share_additional_attributes(self):
+        expected_attribute_name = 'attr'
+        expected_attribute_value = 'value of attr'
+        c1 = Card('9', 'h')
+        c1.attr = expected_attribute_value
+        c2 = Card('9', 'h')
+        self.assertEqual(hasattr(c2, expected_attribute_name), True)
+        self.assertEqual(c2.attr, expected_attribute_value)
+
+if __name__ == "__main__":
+    unittest.main()
+
