Add init incorrect test

adonath · Dec 17, 2024 · 677409b · 677409b
1 parent fab4466
commit 677409b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/test_gmm.py b/tests/test_gmm.py
@@ -52,6 +52,22 @@ def test_create():
     assert gmm.n_parameters == 17
 
 
+def test_init_incorrect():
+    with pytest.raises(ValueError):
+        GaussianMixtureModelJax(
+            means=jnp.zeros((2, 3)),
+            covariances=jnp.zeros((2, 3, 3)),
+            weights=jnp.zeros((2,)),
+        )
+
+    with pytest.raises(ValueError):
+        GaussianMixtureModelJax(
+            means=jnp.zeros((1, 2, 3, 1)),
+            covariances=jnp.zeros((1, 2, 3, 3)),
+            weights=jnp.zeros((1, 1, 4, 1)),
+        )
+
+
 def test_against_sklearn(gmm_jax):
     x = np.array([
         [1, 2, 3],