def test_complex_array(self):
"""Test complex array is correctly formatted"""
A = np.array([[0.453 - 0.543j, 1, 0.213 + 8j],
[-0.543j, 1.342 + 0.5j, 1e-4 - 2e2j]])
res = "\n".join(numpy_to_blackbird(A, "A"))
expected = dedent("""\
complex array A[2, 3] =
0.453-0.543j, 1.0+0.0j, 0.213+8.0j
-0.0-0.543j, 1.342+0.5j, 0.0001-200.0j
""")
assert res == expected
def test_unknown_array(self):
"""Test non-numeric array raises an exception"""
A = np.array([True, False])
with pytest.raises(ValueError, match="unsupported type"):
numpy_to_blackbird(A, "A")
def test_int_array(self):
"""Test integer array is correctly formatted"""
A = np.array([[1, -0, -15]])
res = "\n".join(numpy_to_blackbird(A, "A"))
expected = "int array A[1, 3] =\n 1, 0, -15\n"
assert res == expected
def test_real_array(self):
"""Test real array is correctly formatted"""
A = np.array([[0.453, 1, 0.213], [-0.543, 1.342, 1e-4]])
res = "\n".join(numpy_to_blackbird(A, "A"))
expected = "float array A[2, 3] =\n 0.453, 1.0, 0.213\n -0.543, 1.342, 0.0001\n"
assert res == expected
