Skip to content

Commit 0daf105

Browse files
authored
Merge pull request #210 from SwayamInSync/imag-test
TEST: Tests for `imag` and `real`
2 parents 4af5d64 + d24bd55 commit 0daf105

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

quaddtype/tests/test_quaddtype.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3095,4 +3095,14 @@ def test_buffer():
30953095
buff = a.data
30963096

30973097
reconstructed = np.frombuffer(buff, dtype=QuadPrecDType())[0]
3098-
assert reconstructed == a, "Buffer reconstruction failed"
3098+
assert reconstructed == a, "Buffer reconstruction failed"
3099+
3100+
@pytest.mark.parametrize("value", [0.0, -0.0, 1.0, -1.0, 3.14, -2.71, "inf", "-inf", "nan"])
3101+
def test_imag_real(value):
3102+
a = QuadPrecision(value)
3103+
if np.isnan(a):
3104+
assert np.isnan(a.real), "Real part of NaN should be NaN"
3105+
assert a.imag == QuadPrecision(0.0), "Imaginary part should be zero"
3106+
return
3107+
assert a.real == a, "Real part mismatch"
3108+
assert a.imag == QuadPrecision(0.0), "Imaginary part should be zero"

0 commit comments

Comments
 (0)