aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_enums.py2
-rw-r--r--tests/test_neuralNetwork.py15
2 files changed, 16 insertions, 1 deletions
diff --git a/tests/test_enums.py b/tests/test_enums.py
index 73f4009..21b2057 100644
--- a/tests/test_enums.py
+++ b/tests/test_enums.py
@@ -5,7 +5,7 @@ import unittest
from imago.data.enums import Player
class TestEnums(unittest.TestCase):
- """Test parseHelpers module."""
+ """Test enums module."""
def testOtherPlayer(self):
"""Test method to get the other player"""
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")