Skip to content

Commit 7091f01

Browse files
committed
improve adapter tests
1 parent 49b45d7 commit 7091f01

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_adapters/test_adapters.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def test_simple_transforms(random_data):
9797

9898
result = adapter(random_data)
9999

100-
assert np.isfinite(result["o1"][0, 0])
101-
assert np.isfinite(result["p2"][0, 0])
102-
assert np.isfinite(result["t1"][0, 0])
103-
assert np.isfinite(result["p1"][0, 0])
100+
assert np.array_equal(result["o1"], np.log(random_data["o1"]))
101+
assert np.array_equal(result["p2"], np.log(random_data["p2"]))
102+
assert np.array_equal(result["t1"], np.log1p(random_data["t1"]))
103+
assert np.array_equal(result["p1"], np.sqrt(random_data["p1"]))

0 commit comments

Comments
 (0)