added exercise code

smpotts · smpotts · commit ac224e541928 · 2024-06-06T12:38:41.000-05:00
diff --git a/src/kangaroo.py b/src/kangaroo.py
@@ -0,0 +1,23 @@
+class Kangaroo:
+
+    def __init__(self, name, pouch_contents=None):
+        # __init__ runs, it checks the value of contents and,
+        # if necessary, creates a new empty list.  That way,
+        # every Kangaroo that gets the default value gets a
+        # reference to a different list.
+
+        self.name = name
+
+        if pouch_contents is None:
+            pouch_contents = []
+        self.pouch_contents = pouch_contents
+
+    def put_in_pouch(self, obj):
+        self.pouch_contents.append(obj)
+
+    def __str__(self):
+        t = [self.name + ' has pouch contents:']
+        for thing in self.pouch_contents:
+            s = '    ' + object.__str__(thing)
+            t.append(s)
+        return '\n'.join(t)
diff --git a/tests/test_kangaroo.py b/tests/test_kangaroo.py
@@ -0,0 +1,24 @@
+import unittest
+from src.kangaroo import Kangaroo
+
+
+class TestKangaroo(unittest.TestCase):
+    def test_initialization(self):
+        kangaroo = Kangaroo("kangaroo")
+        self.assertEqual(kangaroo.pouch_contents, [])
+
+    def test_kangaroo_str(self):
+        kangaroo = Kangaroo("kangaroo")
+        kangaroo.put_in_pouch("hello")
+        kangaroo.put_in_pouch("world")
+        print(kangaroo)
+
+    def test_put_in_pouch(self):
+        kanga = Kangaroo("kanga")
+        roo = Kangaroo("roo")
+        kanga.put_in_pouch(roo)
+        print(kanga)
+
+
+if __name__ == '__main__':
+    unittest.main()
