Skip to content

Commit 5e2da0f

Browse files
committed
also test inverse of simple transforms
1 parent 7091f01 commit 5e2da0f

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

tests/test_adapters/test_adapters.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,11 +93,20 @@ def test_simple_transforms(random_data):
9393
# check if simple transforms are applied correctly
9494
from bayesflow.adapters import Adapter
9595

96-
adapter = Adapter().log(["o1", "p2"]).log("t1", p1=True).sqrt("p1")
96+
adapter = Adapter().log(["p2", "t2"]).log("t1", p1=True).sqrt("p1")
9797

9898
result = adapter(random_data)
9999

100-
assert np.array_equal(result["o1"], np.log(random_data["o1"]))
101100
assert np.array_equal(result["p2"], np.log(random_data["p2"]))
101+
assert np.array_equal(result["t2"], np.log(random_data["t2"]))
102102
assert np.array_equal(result["t1"], np.log1p(random_data["t1"]))
103103
assert np.array_equal(result["p1"], np.sqrt(random_data["p1"]))
104+
105+
# inverse results should match the original input
106+
inverse = adapter.inverse(result)
107+
108+
assert np.array_equal(inverse["p2"], random_data["p2"])
109+
assert np.array_equal(inverse["t2"], random_data["t2"])
110+
assert np.array_equal(inverse["t1"], random_data["t1"])
111+
# numerical inaccuries prevent np.array_equal to work here
112+
assert np.allclose(inverse["p1"], random_data["p1"])

0 commit comments

Comments
 (0)