added ex5

2025-01-28 16:52:27 +01:00
parent 68ccc3c183
commit 2fda876df3
3 changed files with 238 additions and 2 deletions
--- a/tests/test_ex5.py
+++ b/tests/test_ex5.py
@@ -1,8 +1,21 @@
 import unittest

+from Ex5 import playMinitrisFastAndWell
+
+
 class MyTestCase(unittest.TestCase):
    def test_simple1(self):
-        pass
+        self.assertEqual(
+            playMinitrisFastAndWell(3, 4, [((1, 0), (0, 0)), ((1, 0), (0, 0)), ((1, 0), (0, 0))]),
+            [[0, 0], [1, 0], [2, 0]]
+        )
+
+    def test_simple2(self):
+        self.assertEqual(
+            playMinitrisFastAndWell(3, 4, [((1, 0), (0, 0)), ((1, 1), (0, 1)), ((1, 1), (0, 0))]),
+            [[1, 0], [0, 0], [2, 0]]
+        )
+

 if __name__ == '__main__':
    unittest.main()