Jupytertest

 avatar
unknown
python
3 years ago
835 B
1
Indexable
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)