Skip to content

Commit c7bc232

Browse files
committed
TST: more type cases in test_pset_hash_recalc_from_df
1 parent 0403e3c commit c7bc232

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

tests/test_all.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -675,9 +675,17 @@ def test_pset_hash_skip_cols():
675675
],
676676
)
677677
def test_pset_hash_recalc_from_df(non_py_type):
678-
params = ps.plist(
679-
"a", [1, "3", np.sin, [], None, 1.23, dummy_func, non_py_type]
678+
stuff = ps.plist(
679+
"stuff", [1, "3", np.sin, [], None, 1.23, dummy_func, non_py_type]
680680
)
681+
plists = [
682+
stuff,
683+
ps.plist("py_int", range(len(stuff))),
684+
ps.plist("py_int_from_np", np.arange(len(stuff)).tolist()),
685+
ps.plist("py_int_intspace", ps.intspace(1, 1000, len(stuff)).tolist()),
686+
ps.plist("py_float_from_np", np.random.rand(len(stuff)).tolist()),
687+
]
688+
params = ps.pgrid([zip(*plists)])
681689
df = ps.run(lambda pset: {}, params, save=False)
682690
for idx, row in df.iterrows():
683691
df.at[idx, "_pset_hash_new"] = ps.pset_hash(row.to_dict())

0 commit comments

Comments
 (0)