Exemple #1
0
 def test_matrix_set(self):
     m1 = Matrix(2, 2)
     m1.setAll(99)
     self.assertEqual(99, m1[0, 0])
     self.assertEqual(99, m1[1, 1])
     m2 = Matrix(2, 2, value=99)
     self.assertEqual(m1, m2)
Exemple #2
0
    def test_matrix_scale(self):
        m = Matrix(2, 2, value=1)
        m.scaleColumn(0, 2)
        self.assertEqual(2, m[0, 0])
        self.assertEqual(2, m[1, 0])

        m.setAll(1)
        m.scaleRow(1, 2)
        self.assertEqual(2, m[1, 0])
        self.assertEqual(2, m[1, 1])

        with self.assertRaises(IndexError):
            m.scaleColumn(10, 99)

        with self.assertRaises(IndexError):
            m.scaleRow(10, 99)