Initial commit

1 files changed, 27 insertions, 0 deletions

diff --git a/21/test.py b/21/test.py
new file mode 100755
index 0000000..4527f49
--- /dev/null
+++ b/21/test.py
@@ -0,0 +1,27 @@
+#!/usr/bin/python
+
+import unittest
+
+import main
+
+
+
+class ExamplesPart1(unittest.TestCase):
+    def test_example1(self):
+        player = main.Entity()
+        player.health = 8
+        player.damage = 0
+        player.armour = 0
+
+        boss = main.Entity()
+        boss.health = 12
+        boss.damage =  7
+        boss.armour =  2
+
+        result = main.solve_partn(1, player, boss)
+        self.assertEqual(result, 40 + 25)
+
+
+
+if __name__ == "__main__":
+    unittest.main()