Add basic workflow tests and reduce coupling depth for tests

Author: stefanradev93

tests/test_networks/conftest.py

@@ -38,14 +38,14 @@ def flow_matching_subnet(subnet):
 def coupling_flow():
     from bayesflow.networks import CouplingFlow
 
-    return CouplingFlow()
+    return CouplingFlow(depth=2)
 
 
 @pytest.fixture()
 def coupling_flow_subnet(subnet):
     from bayesflow.networks import CouplingFlow
 
-    return CouplingFlow(subnet=subnet)
+    return CouplingFlow(depth=2, subnet=subnet)
 
 
 @pytest.fixture()

tests/test_workflows/conftest.py

@@ -0,0 +1,15 @@
+import pytest
+
+
+@pytest.fixture()
+def inference_network():
+    from bayesflow.networks import CouplingFlow
+
+    return CouplingFlow(depth=2)
+
+
+@pytest.fixture()
+def summary_network():
+    from bayesflow.networks import TimeSeriesTransformer
+
+    return TimeSeriesTransformer(embed_dims=(8, 8), mlp_widths=(32, 32), mlp_depths=(1, 1))

tests/test_workflows/test_basic_workflow.py

@@ -0,0 +1,21 @@
+import bayesflow as bf
+
+
+def test_classifier_two_sample_test(inference_network, summary_network):
+    workflow = bf.BasicWorkflow(
+        inference_network=inference_network,
+        summary_network=summary_network,
+        inference_variables=["parameters"],
+        summary_variables=["observables"],
+        simulator=bf.simulators.SIR(),
+    )
+
+    history = workflow.fit_online(epochs=2, batch_size=32, num_batches_per_epoch=2)
+    plots = workflow.plot_default_diagnostics(test_data=50, num_samples=50)
+    metrics = workflow.compute_default_diagnostics(test_data=50, num_samples=50, variable_names=["p1", "p2"])
+
+    assert "loss" in list(history.history.keys())
+    assert len(history.history["loss"]) == 2
+    assert list(plots.keys()) == ["losses", "recovery", "calibration_ecdf", "z_score_contraction"]
+    assert list(metrics.columns) == ["p1", "p2"]
+    assert metrics.values.shape == (3, 2)