Example #1
0
def test_size():
    a = MyMatrix([[1, 2, 5], [3, 4, 6]])
    assert a.size() == (2, 3)
    a = MyMatrix([[], []])
    assert a.size() == (2, 0)
    a = MyMatrix([])
    assert a.size() == (0, 0)
Example #2
0
def test_transposed():
    a = MyMatrix([[1, 3, 4], [5, 7, 2]])
    assert (a.transposed().get_data() == [[1, 5], [3, 7], [4, 2]])
    assert (a.get_data() == [[1, 3, 4], [5, 7, 2]])
    a = MyMatrix([[1, 3], [5, 7], [9, 0]])
    assert (a.transposed().get_data() == [[1, 5, 9], [3, 7, 0]])
    assert (a.get_data() == [[1, 3], [5, 7], [9, 0]])
Example #3
0
def test_transpose():
    a = MyMatrix([[1, 2, 5], [3, 4, 6]])
    a.transpose()
    assert (a.matrix == [[1, 3], [2, 4], [5, 6]])
Example #4
0
def test_flipped_up_down():
    a = MyMatrix([[1, 3, 4], [5, 7, 2]])
    assert (a.flipped_up_down().get_data() == [[5, 7, 2], [1, 3, 4]])
    assert (a.get_data() == [[1, 3, 4], [5, 7, 2]])
Example #5
0
def test_repr():
    a = MyMatrix([[1, 2, 5], [3, 4, 6]])
    assert a.__repr__() == ' 1 2 5\n 3 4 6\n'
Example #6
0
def test_flipped_left_right():
    a = MyMatrix([[1, 3, 4], [5, 7, 2]])
    assert (a.flipped_left_right().get_data() == [[4, 3, 1], [2, 7, 5]])
    assert (a.get_data() == [[1, 3, 4], [5, 7, 2]])
Example #7
0
def test_flip_up_down():
    a = MyMatrix([[1, 2, 5], [3, 4, 6]])
    a.flip_up_down()
    assert a.matrix == [[3, 4, 6], [1, 2, 5]]
Example #8
0
def test_flip_left_right():
    a = MyMatrix([[1, 2, 5], [3, 4, 6]])
    a.flip_left_right()
    assert (a.matrix == [[5, 2, 1], [6, 4, 3]])