"""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")