tests/codewars/test_who_the_most_money.py

@@ -0,0 +1,15 @@
+from src.codewars.kata_who_the_most_money import Student, most_money
+
+
+def test_most_money():
+    phil = Student("Phil", 2, 2, 1)
+    cam = Student("Cameron", 2, 2, 0)
+    geoff = Student("Geoff", 0, 3, 0)
+
+    assert most_money([cam, geoff, phil]) == "Phil"
+
+    phil = Student("Phil", 2, 2, 2)
+    cam = Student("Cameron", 2, 2, 2)
+    geoff = Student("Geoff", 2, 2, 2)
+
+    assert most_money([cam, geoff, phil]) == "all"