Update assertions in test

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
bystrogenomics · Jul 18, 2024 · 835016c · 835016c
1 parent 22be7bf
commit 835016c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/python/bystro/parent_of_origin/tests/test_parent_of_origin.py b/python/python/bystro/parent_of_origin/tests/test_parent_of_origin.py
@@ -257,3 +257,5 @@ def test_multi2_fit():
     )
     model = POEMultipleSNP2(n_repeats=10)
     model.fit(data["phenotypes"], data["genotypes"], seed=2021)
+    assert model is not None, "Model fitting failed"
+    assert isinstance(model, POEMultipleSNP2), "Model type is incorrect"