class TestFunctions(unittest.TestCase):
"""
TODO: check if test is running
"""
def shouldReturnIdealFunctions(self):
expectedIdealY1 = 'y33'
expectedIdealY2 = 'y10'
expectedIdealY3 = 'y48'
expectedIdealY4 = 'y18'
Ideal_Y1 = ideal_function(Train_Y1, Ideal_data)
Ideal_Y2 = ideal_function(Train_Y2, Ideal_data)
Ideal_Y3 = ideal_function(Train_Y3, Ideal_data)
Ideal_Y4 = ideal_function(Train_Y4, Ideal_data)
self.assertEqual(expectedIdealY1, IdealY1)
self.assertEqual(expectedIdealY1, IdealY2)
self.assertEqual(expectedIdealY1, IdealY3)
self.assertEqual(expectedIdealY1, IdealY4)
if __name__ == '__main__':
unittest.main(argv=['first-arg-is-ignored'], exit=False)