Skip to content

Commit d24bda1

Browse files
committed
Tests: check LoadShape.UseFloat64
1 parent 78552a3 commit d24bda1

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/test_alt.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,10 +139,16 @@
139139

140140

141141
if isinstance(c, LoadShapeObjMixin):
142+
c.PMult = c.PMult * 100000000 # force it to lose precision
142143
mult_f64 = c.PMult
143144
c.UseFloat32()
144145
mult_f32 = c.PMult
145146
nptest.assert_allclose(mult_f32, mult_f64)
147+
assert list(mult_f32) != list(mult_f64)
148+
c.UseFloat64()
149+
mult_f64_2 = c.PMult
150+
assert list(mult_f64_2) != list(mult_f64)
151+
assert list(mult_f32) == list(mult_f64_2)
146152

147153

148154
if isinstance(c, TransformerObjMixin):

0 commit comments

Comments
 (0)