|
1 | 1 | import numpy as np
|
| 2 | +import pytest |
2 | 3 |
|
3 | 4 | from caput import truncate
|
4 | 5 |
|
@@ -58,21 +59,23 @@ def test_truncate_float():
|
58 | 59 | assert truncate.bit_truncate_double(32.121, -1) == 0
|
59 | 60 | assert truncate.bit_truncate_float(32.121, np.inf) == 0
|
60 | 61 | assert truncate.bit_truncate_double(32.121, np.inf) == 0
|
61 |
| - assert truncate.bit_truncate_float(32.121, np.nan) == 0 |
62 |
| - assert truncate.bit_truncate_double(32.121, np.nan) == 0 |
63 | 62 | assert truncate.bit_truncate_float(np.inf, 1) == np.inf
|
64 | 63 | assert truncate.bit_truncate_double(np.inf, 1) == np.inf
|
65 | 64 | assert np.isnan(truncate.bit_truncate_float(np.nan, 1))
|
66 | 65 | assert np.isnan(truncate.bit_truncate_double(np.nan, 1))
|
| 66 | + |
67 | 67 | assert truncate.bit_truncate_float(np.inf, np.inf) == 0
|
68 | 68 | assert truncate.bit_truncate_double(np.inf, np.inf) == 0
|
69 |
| - assert truncate.bit_truncate_float(np.inf, np.nan) == 0 |
70 |
| - assert truncate.bit_truncate_double(np.inf, np.nan) == 0 |
71 |
| - assert truncate.bit_truncate_float(np.nan, np.nan) == 0 |
72 |
| - assert truncate.bit_truncate_double(np.nan, np.nan) == 0 |
73 | 69 | assert truncate.bit_truncate_float(np.nan, np.inf) == 0
|
74 | 70 | assert truncate.bit_truncate_double(np.nan, np.inf) == 0
|
75 | 71 |
|
| 72 | + # Test that an error is raised when `err` is `NaN` |
| 73 | + with pytest.raises(ValueError): |
| 74 | + truncate.bit_truncate_float(32.121, np.nan) |
| 75 | + |
| 76 | + with pytest.raises(ValueError): |
| 77 | + truncate.bit_truncate_double(32.121, np.nan) |
| 78 | + |
76 | 79 |
|
77 | 80 | def test_truncate_array():
|
78 | 81 | assert (
|
|
0 commit comments