import unittest

from Ex5 import playMinitrisFastAndWell


class MyTestCase(unittest.TestCase):
    def test_simple1(self):
        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()