File tree Expand file tree Collapse file tree 1 file changed +11
-2
lines changed Expand file tree Collapse file tree 1 file changed +11
-2
lines changed Original file line number Diff line number Diff 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" ])
You can’t perform that action at this time.
0 commit comments