1 files changed, 15 insertions, 0 deletions
diff --git a/tests/test_neuralNetwork.py b/tests/test_neuralNetwork.py
new file mode 100644
index 0000000..dfcbd7a
--- /dev/null
+++ b/tests/test_neuralNetwork.py
@@ -0,0 +1,15 @@
+"""Tests for neural network module."""
+
+import unittest
+
+from imago.engine.keras.neuralNetwork import NeuralNetwork
+
+class TestNeuralNetwork(unittest.TestCase):
+    """Test neural network module."""
+
+    def testLoadBaseClass(self):
+        """Test error when creating model with the base NeuralNetwork class"""
+
+        self.assertRaises(NotImplementedError,
+                          NeuralNetwork,
+                          "non/existing/file")