Skip to content

Commit 65fdbd2

Browse files
committed
Update stancsv tests to np.float64
1 parent 54a695f commit 65fdbd2

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

test/test_stancsv.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,11 @@ def test_csv_bytes_to_numpy_no_header():
2323
[-6.85511, 0.994945, 0.787025, 2, 3, 0, 6.85536, 0.310589],
2424
[-6.85511, 0.812189, 0.787025, 1, 1, 0, 7.16517, 0.310589],
2525
],
26-
dtype=np.float32,
26+
dtype=np.float64,
2727
)
2828
arr_out = stancsv.csv_bytes_list_to_numpy(lines, includes_header=False)
2929
assert np.array_equal(arr_out, expected)
30-
assert arr_out[0].dtype == np.float32
30+
assert arr_out[0].dtype == np.float64
3131

3232

3333
def test_csv_bytes_to_numpy_no_header_no_polars():
@@ -44,12 +44,12 @@ def test_csv_bytes_to_numpy_no_header_no_polars():
4444
[-6.85511, 0.994945, 0.787025, 2, 3, 0, 6.85536, 0.310589],
4545
[-6.85511, 0.812189, 0.787025, 1, 1, 0, 7.16517, 0.310589],
4646
],
47-
dtype=np.float32,
47+
dtype=np.float64,
4848
)
4949
with mock.patch.dict("sys.modules", {"polars": None}):
5050
arr_out = stancsv.csv_bytes_list_to_numpy(lines, includes_header=False)
5151
assert np.array_equal(arr_out, expected)
52-
assert arr_out[0].dtype == np.float32
52+
assert arr_out[0].dtype == np.float64
5353

5454

5555
def test_csv_bytes_to_numpy_with_header():
@@ -70,7 +70,7 @@ def test_csv_bytes_to_numpy_with_header():
7070
[-6.85511, 0.994945, 0.787025, 2, 3, 0, 6.85536, 0.310589],
7171
[-6.85511, 0.812189, 0.787025, 1, 1, 0, 7.16517, 0.310589],
7272
],
73-
dtype=np.float32,
73+
dtype=np.float64,
7474
)
7575
arr_out = stancsv.csv_bytes_list_to_numpy(lines, includes_header=True)
7676
assert np.array_equal(arr_out, expected)
@@ -84,7 +84,7 @@ def test_csv_bytes_to_numpy_single_element():
8484
[
8585
[-6.76206],
8686
],
87-
dtype=np.float32,
87+
dtype=np.float64,
8888
)
8989
arr_out = stancsv.csv_bytes_list_to_numpy(lines, includes_header=False)
9090
assert np.array_equal(arr_out, expected)
@@ -98,7 +98,7 @@ def test_csv_bytes_to_numpy_single_element_no_polars():
9898
[
9999
[-6.76206],
100100
],
101-
dtype=np.float32,
101+
dtype=np.float64,
102102
)
103103
with mock.patch.dict("sys.modules", {"polars": None}):
104104
arr_out = stancsv.csv_bytes_list_to_numpy(lines, includes_header=False)
@@ -123,7 +123,7 @@ def test_csv_bytes_to_numpy_with_header_no_polars():
123123
[-6.85511, 0.994945, 0.787025, 2, 3, 0, 6.85536, 0.310589],
124124
[-6.85511, 0.812189, 0.787025, 1, 1, 0, 7.16517, 0.310589],
125125
],
126-
dtype=np.float32,
126+
dtype=np.float64,
127127
)
128128
with mock.patch.dict("sys.modules", {"polars": None}):
129129
arr_out = stancsv.csv_bytes_list_to_numpy(lines, includes_header=True)
@@ -233,7 +233,7 @@ def test_parsing_adaptation_lines_dense():
233233
[0.230843, 3.92459, 0.126989],
234234
[0.0509365, 0.126989, 3.82718],
235235
],
236-
dtype=np.float32,
236+
dtype=np.float64,
237237
)
238238
assert step_size == 0.775147
239239
assert mass_matrix is not None

0 commit comments

Comments
 (0)