diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_enums.py | 2 | ||||
-rw-r--r-- | tests/test_neuralNetwork.py | 15 |
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") |