def test_2_by_2(self): m1 = [[1,2], [3,4]] m2 = [[5,6], [7,8]] r = [[19, 22], [43, 50]] self.assertEqual(strassen_multiply(m1, m2), r)
def test_4_by_4(self): m1 = [[ 1, 2, 3, 4], [ 5, 6, 7, 8], [ 9,10,11,12], [13,14,15,16]] m2 = [[17,18,19,20], [21,22,23,24], [25,26,27,28], [29,30,31,32]] r = [[ 250, 260, 270, 280], [ 618, 644, 670, 696], [ 986,1028,1070,1112], [1354,1412,1470,1528]] self.assertEqual(strassen_multiply(m1, m2), r)