예제 #1
0
def test_same_3():
    # 1-D rows
    A = sp.array([[1., 0, 0]])

    B = sp.array([[1., 0, 0]])

    assert (same_rows(A, B) is True)
예제 #2
0
def test_elements_4():
    # 0-D numpy arrays, with different elements
    A = sp.array([1., 0, 0])

    B = sp.array([1, 2, -3])

    assert (same_rows(A, B) is False)
예제 #3
0
def test_shape_7():
    # both row vectors, same elements
    A = sp.array([[1., 0, 0, 0, 0]])

    B = sp.array([[1., 0, 0, 0, 0]])

    assert (same_rows(A, B) is True)
예제 #4
0
def test_shape_8():
    # one row vectors, one column vector
    A = sp.array([[1., 0, 0, 0, 0]])

    B = sp.array([[1., 0, 0, 0, 0]]).T

    assert (same_rows(A, B) is False)
예제 #5
0
def test_shape_6():
    # both row vectors, different elements
    A = sp.array([[1., 0, 0, 0, 0]])

    B = sp.array([[1., 0, 0, 1, 0]])

    assert (same_rows(A, B) is False)
예제 #6
0
def test_shape_5():
    # both column vectors, same elements
    A = sp.array([[1., 2, 3, 4, 5]]).T

    B = sp.array([[1., 2, 3, 4, 5]]).T

    assert (same_rows(A, B) is True)
예제 #7
0
def test_shape_4():
    # both column vectors, different elements
    A = sp.array([[1., 0, 0, 0, 0]]).T

    B = sp.array([[1., 0, 0, 1, 0]]).T

    assert (same_rows(A, B) is False)
예제 #8
0
def test_same_4():
    # identical copies
    A = sp.array([[1., 0, 0],
                  [0, 1, 2],
                  [0, 1, 3]])

    B = A

    assert (same_rows(A, B) is True)
예제 #9
0
def test_elements_5():
    # same shape, just the negative of one another
    A = sp.array([[1., 0, 0],
                  [0, 1, 2],
                  [0, 1, 3]])

    B = -A

    assert (same_rows(A, B) is False)
예제 #10
0
def test_elements_3():
    # same shape, just transposed
    A = sp.array([[1., 0, 0],
                  [0, 1, 2],
                  [0, 1, 3]])

    B = A.T

    assert (same_rows(A, B) is False)
예제 #11
0
def test_2():
    # 1-D nullspace
    A = sp.array([[1., 0, 0],
                  [0, 1, 0]])

    N = artools.nullspace(A)
    N_ref = sp.array([[0., 0, 1]]).T

    assert (artools.same_rows(N, N_ref) is True)
예제 #12
0
def test_shape_1():
    # different shapes
    A = sp.array([[1., 0, 0],
                  [0, 1, 2],
                  [0, 1, 3]])

    B = sp.array([[1., 0, 0],
                  [0, 1, 2]])

    assert (same_rows(A, B) is False)
예제 #13
0
def test_same_1():
    # identical matrices
    A = sp.array([[1., 0, 0],
                  [0, 1, 2],
                  [0, 1, 3]])

    B = sp.array([[1., 0, 0],
                  [0, 1, 2],
                  [0, 1, 3]])

    assert (same_rows(A, B) is True)
예제 #14
0
def test_elements_2():
    # same shape, different elements
    A = sp.array([[1., 0, 0],
                  [0, 1, 2],
                  [0, 1, 3]])

    B = sp.array([[0., 0, 0],
                  [0, 0, 0],
                  [0, 0, 0]])

    assert (same_rows(A, B) is False)
예제 #15
0
def test_same_2():
    # same rows, different order
    A = sp.array([[1., 0, 0],
                  [0, 1, 2],
                  [0, 1, 3]])

    B = sp.array([[1., 0, 0],
                  [0, 1, 3],
                  [0, 1, 2]])

    assert (same_rows(A, B) is True)
예제 #16
0
def test_2D_1():
    # 2-D mass balance triangle
    A = sp.array([[-1., 0],
                  [0, -1],
                  [1, 1]])

    b = sp.array([[0., 0, 1]]).T

    vs = artools.con2vert(A, b)
    vs_ref = sp.array([[0., 1], [0, 0], [1, 0]])

    assert (artools.same_rows(vs, vs_ref) is True)
예제 #17
0
def test_2D_3():
    # mass blanace triangle cut at y = 0.5
    A = sp.array([[-1., 0],
                  [0, -1],
                  [1, 1],
                  [0, 1]])

    b = sp.array([[0., 0, 1, 0.5]]).T

    vs = artools.con2vert(A, b)
    vs_ref = sp.array([[0, 0.5],
                       [0, 0],
                       [1, 0],
                       [0.5, 0.5]])

    assert (artools.same_rows(vs, vs_ref) is True)
예제 #18
0
def test_2D_2():
    # 2-D unit square
    A = sp.array([[-1., 0],
                  [0, -1],
                  [1, 0],
                  [0, 1]])

    b = sp.array([[0., 0, 1, 1]]).T

    vs = artools.con2vert(A, b)
    vs_ref = sp.array([[0., 1],
                       [0, 0],
                       [1, 0],
                       [1, 1]])

    assert (artools.same_rows(vs, vs_ref) is True)
예제 #19
0
def test_arrays_3():
    # A is a column vector, B is a 0-D array
    A = sp.array([[1., 2, 3, 4, 5]]).T
    B = sp.array([1., 2, 3, 4, 5])

    assert (same_rows(A, B) is False)
예제 #20
0
def test_arrays_1():
    # both 0-D arrays
    A = sp.array([1., 2, 3, 4, 5])
    B = sp.array([1., 2, 3, 4, 5])

    assert (same_rows(A, B) is True)
예제 #21
0
def test_arrays_2():
    # A is a row vector, B is a 0-D array
    A = sp.array([[1., 2, 3, 4, 5]])
    B = sp.array([1., 2, 3, 4, 5])

    assert (same_rows(A, B) is False)