Stub testcase for model.cond

scverse · Apr 2, 2024 · 90e8247 · 90e8247
1 parent 3a1ef9a
commit 90e8247
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/multi_condition_comparisions/_util/formulaic.py b/src/multi_condition_comparisions/_util/formulaic.py
@@ -8,6 +8,7 @@
 from formulaic.materializers import PandasMaterializer
 from formulaic.materializers.types import EvaluatedFactor
 from interface_meta import override
+from formulaic.parser.types import Factor
 
 
 @dataclass

diff --git a/tests/test_base.py b/tests/test_base.py
@@ -17,3 +17,18 @@ def fit(self, **kwargs) -> None:
 
         def _test_single_contrast(self, contrast: Sequence[float], **kwargs) -> DataFrame:
             pass
+
+    return _MockLinearModel
+
+
+@pytest.mark.parametrize(
+    "formula,cond_kwargs,expected_contrast",
+    [
+        ["~ condition", {"condition": "A"}, [0, 0]],
+    ],
+)
+def test_model_cond(test_adata_minimal, MockLinearModel, formula, cond_kwargs, expected_contrast):
+    mod = MockLinearModel(test_adata_minimal, formula)
+    actual_contrast = mod.cond(**cond_kwargs)
+    assert actual_contrast.tolist() == expected_contrast
+    assert actual_contrast.index.tolist() == mod.design.columns.tolist()