Python MatrixCalculator.get_cols Examples

Example #1

File: matrix_calculator_test.py Project: lmatias85/Python-MatrixCalculator

def test_mat_cols_2():
    a = [[0, 1, 1, 1, 1], [1, 0, 1, 1, 1], [1, 1, 2, 1, 0], [1, 1, 1, 1, 0],
         [1, 1, 1, 1, 0]]
    assert calc.get_cols(a, [3, 4]) == [[1, 1, 1, 1, 1], [1, 1, 0, 0, 0]]

Example #2

File: matrix_calculator_test.py Project: lmatias85/Python-MatrixCalculator

def test_mat_cols_1():
    a = [[1, 2, 1], [1, -1, 2], [0, 1, 0]]
    assert calc.get_cols(a, [1, 0]) == [[2, -1, 1], [1, 1, 0]]