Ejemplo n.º 1
0
def test_outer_2():
    vector_1 = [[1, 0, 0]]
    vector_2 = [[0, 0, 5]]
    ar, ia, ja = outer_product(vector_1, vector_2)
    assert ar == [5]
    assert ia == [0, 1, 1, 1]
    assert ja == [2]
Ejemplo n.º 2
0
def test_outer_3():
    vector_1 = [[3, 0, 5]]
    vector_2 = [[3, 0, 5]]
    ar, ia, ja = outer_product(vector_1, vector_2)
    assert ar == [9, 15, 15, 25]
    assert ia == [0, 2, 2, 4]
    assert ja == [0, 2, 0, 2]
Ejemplo n.º 3
0
def test_outer_1():
    vector_1 = [[1, 0, 3]]
    vector_2 = [[3, 0, 5]]
    ar, ia, ja = outer_product(vector_1, vector_2)
    assert ar == [3, 5, 9, 15]
    assert ia == [0, 2, 2, 4]
    assert ja == [0, 2, 0, 2]
Ejemplo n.º 4
0
def test_outer_wrong_vector_format_1():
    vector_1 = [[1], [0], [0]]
    vector_2 = [[0], [0], [5]]
    with pytest.raises(ValueError, match=r"Both vectors must be 1xn size."):
        outer_product(vector_1, vector_2)