예제 #1
0
def test_create_empty_matrix_vector_complex():
    output = np.array([0. + 0.j, 0. + 0.j, 0. + 0.j, 0. + 0.j, 0. + 0.j])
    value_array = matrixfun.create_empty_matrix(5, complex_vals=True)
    assert np.array_equal(output, value_array)
예제 #2
0
def test_create_empty_matrix_vector_float():
    output = np.array([0., 0., 0., 0., 0.])
    value_array = matrixfun.create_empty_matrix(5)
    assert np.array_equal(output, value_array)
예제 #3
0
def test_create_empty_matrix_empty_float():
    output = np.array([], dtype=np.complex128)
    value_array = matrixfun.create_empty_matrix(0)
    assert np.array_equal(output, value_array)
예제 #4
0
def test_create_empty_matrix_empty_complex():
    output = np.array([], dtype=np.float64)
    value_array = matrixfun.create_empty_matrix(0, complex_vals=True)
    assert np.array_equal(output, value_array)
예제 #5
0
def test_create_empty_matrix_floats():
    output = np.array([[0., 0., 0., 0.], [0., 0., 0., 0.], [0., 0., 0., 0.]])
    value_array = matrixfun.create_empty_matrix((3, 4))
    assert np.array_equal(output, value_array)