def test_parse_complex(): assert parse_complex('0') == 0 assert parse_complex('1') == 1 assert parse_complex('i') == 1j assert parse_complex('2i') == 2j assert parse_complex('-i') == -1j assert parse_complex('+i') == 1j assert parse_complex('1 + i - i') == 1 assert parse_complex('1 + 2i - 3 i') == 1 - 1j np.testing.assert_allclose(parse_complex('exp 2'), np.e**2, atol=1e-8)
def test_parse_complex_raw_cases_from_quirk(): assert parse_complex("0") == 0 assert parse_complex("1") == 1 assert parse_complex("-1") == -1 assert parse_complex("i") == 1j assert parse_complex("-i") == -1j assert parse_complex("2") == 2 assert parse_complex("2i") == 2j assert parse_complex("-2i") == -2j assert parse_complex("3-2i") == 3 - 2j assert parse_complex("1-i") == 1 - 1j assert parse_complex("1+i") == 1 + 1j assert parse_complex("-5+2i") == -5 + 2j assert parse_complex("-5-2i") == -5 - 2j assert parse_complex("3/2i") == 1.5j assert parse_complex("\u221A2-\u2153i") == np.sqrt(2) - 1j / 3 assert parse_complex("1e-10") == 0.0000000001 assert parse_complex("1e+10") == 10000000000 assert parse_complex("2.5e-10") == 0.00000000025 assert parse_complex("2.5E-10") == 0.00000000025 assert parse_complex("2.5e+10") == 25000000000 assert parse_complex("2.e+10") == 20000000000 np.testing.assert_allclose(parse_complex("e"), np.e) np.testing.assert_allclose(parse_complex("pi e"), np.pi * np.e) np.testing.assert_allclose(parse_complex("pi e 2"), np.pi * np.e * 2) np.testing.assert_allclose(parse_complex("2 pi"), 2 * np.pi)
def test_parse_complex_expression_cases_from_quirk(): np.testing.assert_allclose(parse_complex("1/3"), 1 / 3) np.testing.assert_allclose(parse_complex("2/3/5"), (2 / 3) / 5) np.testing.assert_allclose(parse_complex("2/3/5*7/13"), ((((2 / 3) / 5)) * 7) / 13) np.testing.assert_allclose(parse_complex("2-3-5"), -6) np.testing.assert_allclose(parse_complex("1/3+2i"), 1 / 3 + 2j) np.testing.assert_allclose(parse_complex("(1/3)+2i"), 1 / 3 + 2j) np.testing.assert_allclose(parse_complex("1/(3+2i)"), 1 / (3 + 2j), atol=1e-8) np.testing.assert_allclose(parse_complex("1/sqrt(3+2i)"), 1 / ((3 + 2j)**0.5), atol=1e-8) np.testing.assert_allclose(parse_complex("i^i"), 0.20787957635076193, atol=1e-8) np.testing.assert_allclose(parse_complex("√i"), np.sqrt(1j)) np.testing.assert_allclose(parse_complex("√4i"), 2j) np.testing.assert_allclose(parse_complex("sqrt4i"), 2j) np.testing.assert_allclose(parse_complex("sqrt√4i"), np.sqrt(2) * 1j) np.testing.assert_allclose(parse_complex("sqrt√4-i"), np.sqrt(2) - 1j) np.testing.assert_allclose(parse_complex("----------1"), 1) np.testing.assert_allclose(parse_complex("---------1"), -1) np.testing.assert_allclose(parse_complex("---+--+--1"), -1) np.testing.assert_allclose(parse_complex("0---+--+--1"), -1) np.testing.assert_allclose(parse_complex("0---+--+--1*"), -1) np.testing.assert_allclose(parse_complex("2+3^"), 5) np.testing.assert_allclose(parse_complex("cos(45) + i sin(45)"), np.sqrt(0.5) + 1j * np.sqrt(0.5), atol=1e-8) np.testing.assert_allclose(parse_complex("cos(45) + i (sin 45)"), np.sqrt(0.5) + 1j * np.sqrt(0.5), atol=1e-8) np.testing.assert_allclose(parse_complex("e^(pi i)"), -1, atol=1e-8) np.testing.assert_allclose(parse_complex("exp(ln(2))"), 2, atol=1e-8) np.testing.assert_allclose(parse_complex("sin(arcsin(0.5))"), 0.5, atol=1e-8) np.testing.assert_allclose(parse_complex("cos(arccos(0.5))"), 0.5, atol=1e-8) np.testing.assert_allclose(parse_complex("sin(asin(0.5))"), 0.5, atol=1e-8) np.testing.assert_allclose(parse_complex("cos(acos(0.5))"), 0.5, atol=1e-8)