added test for the model config util

cpmpercussion · Jun 26, 2024 · 6f3e0e5 · 6f3e0e5
1 parent 0a5fd9a
commit 6f3e0e5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/impsy/tests/test_model.py b/impsy/tests/test_model.py
@@ -41,3 +41,9 @@ def test_training():
     Xs, ys = train.seq_to_overlapping_format(slices)
     history = net.train(Xs, ys, num_epochs=num_epochs, saving=False)
     assert isinstance(history, tf.keras.callbacks.History)
+
+
+def test_model_config():
+    """Tests the model config function."""
+    conf = utils.mdrnn_config('s')
+    assert(conf["units"] == 64)