From 6f3e0e54278bda51d3a65d784a75d8e7c75ce7f2 Mon Sep 17 00:00:00 2001
From: Charles Martin <cpm@charlesmartin.com.au>
Date: Wed, 26 Jun 2024 11:43:36 +1000
Subject: [PATCH] added test for the model config util

---
 impsy/tests/test_model.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/impsy/tests/test_model.py b/impsy/tests/test_model.py
index f493045..75f1576 100644
--- 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)