1 files changed, 15 insertions, 0 deletions

diff --git a/day7/test_init.py b/day7/test_init.py
new file mode 100644
index 0000000..ddcf814
--- /dev/null
+++ b/day7/test_init.py
@@ -0,0 +1,15 @@
+from day7 import AssignmentTwo
+
+__all__ = []
+
+def test_calculate_cost_assignment_two():
+    assert AssignmentTwo.calculate_cost(16, 5) == 66
+    assert AssignmentTwo.calculate_cost(1, 5) == 10 
+    assert AssignmentTwo.calculate_cost(2, 5) == 6 
+    assert AssignmentTwo.calculate_cost(0, 5) == 15 
+    assert AssignmentTwo.calculate_cost(4, 5) == 1 
+    assert AssignmentTwo.calculate_cost(2, 5) == 6 
+    assert AssignmentTwo.calculate_cost(7, 5) == 3 
+    assert AssignmentTwo.calculate_cost(1, 5) == 10 
+    assert AssignmentTwo.calculate_cost(2, 5) == 6 
+    assert AssignmentTwo.calculate_cost(14, 5) == 45