def test_add_path_big(): prev_cut = path.matrix_cut(test_matrix, 7) next_cut = path.matrix_cut(test_matrix, 8) assert path.add_path(prev_cut, next_cut) == [956 + 111, 37 + 121]
def test_add_path_small(): prev_cut = path.matrix_cut(test_matrix, 1) next_cut = path.matrix_cut(test_matrix, 2) assert path.add_path(prev_cut, next_cut) == [673 + 131, 201 + 131]
def test_matrix_cut_big(): assert path.matrix_cut(test_matrix, 7) == [111, 121, 524]
def test_matrix_cut_over(): with pytest.raises(ValueError): path.matrix_cut(test_matrix, 16)
def test_matrix_cut_small(): assert path.matrix_cut(test_matrix, 2) == [673, 201]