Skip to content

Commit 0df388a

Browse files
committed
[Test] Adds options for pickle
1 parent 0cf2417 commit 0df388a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_no_regression.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def setup_test(pb_name: str, rsolver: str, run_type: str) -> None:
3434
filename = f"{pb_name}_{rsolver}" # Base filename w/o extension. Ex: sod_x_hll, sod_y_hllc...
3535
ref_path = REF_DIR / f"{filename}.npz"
3636

37-
with np.load(ref_path) as ref_data:
37+
with np.load(ref_path, allow_pickle=True) as ref_data:
3838
ref_density = ref_data['rho']
3939
ref_velocity_x = ref_data['u']
4040
ref_velocity_y = ref_data['v']
@@ -44,7 +44,7 @@ def setup_test(pb_name: str, rsolver: str, run_type: str) -> None:
4444
h5_to_numpy(str(FV2D_TMP_DIR / f"{filename}.h5"), str(FV2D_TMP_DIR / f"{filename}.npz"))
4545
test_path = FV2D_TMP_DIR / f"{filename}.npz"
4646

47-
with np.load(test_path) as test_data:
47+
with np.load(test_path, allow_pickle=True) as test_data:
4848
test_density = test_data['rho']
4949
test_velocity_x = test_data['u']
5050
test_velocity_y = test_data['v']

0 commit comments

Comments
 (0)