dataset/problem_10/test.py

import unittest
from dataset.problem_10.buggy import rotate_90_clockwise


class TestRotateMatrix(unittest.TestCase):
    def test_2x2(self):
        matrix = [[1, 2], [3, 4]]
        self.assertEqual(rotate_90_clockwise(matrix), [[3, 1], [4, 2]])


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