30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
import unittest
|
|
import numpy as np
|
|
|
|
import src.Functions
|
|
|
|
class TestFunctions(unittest.TestCase):
|
|
|
|
def testcalcMeanSquaredError(self):
|
|
a = np.array([1, 2, 3, 4, 5])
|
|
b = np.array([1.1, 2.05, 2.95, 4.01, 4.5])
|
|
self.assertEqual(0.05, src.Functions.calcMeanSquaredError(a, b))
|
|
|
|
a = np.array(['1', '2', '3', '4', '5'])
|
|
b = np.array(['1.1', '2.05', '2.95', '4.01', '4.5'])
|
|
self.assertEqual(0.05, src.Functions.calcMeanSquaredError(a, b))
|
|
|
|
def testGetRandomLayer(self):
|
|
layer = (1, 3)
|
|
self.assertEqual(layer, np.shape(src.Functions.getRandomLayer(layer)))
|
|
layer = (3, 1)
|
|
self.assertEqual(layer, np.shape(src.Functions.getRandomLayer(layer)))
|
|
layer = (8, 2)
|
|
self.assertEqual(layer, np.shape(src.Functions.getRandomLayer(layer)))
|
|
layer = (100, 1)
|
|
self.assertEqual(layer, np.shape(src.Functions.getRandomLayer(layer)))
|
|
layer = (1, 1)
|
|
self.assertEqual(layer, np.shape(src.Functions.getRandomLayer(layer)))
|
|
layer = (4, 50)
|
|
self.assertEqual(layer, np.shape(src.Functions.getRandomLayer(layer)))
|