def test_multiple_meas_prompt(self, mock): """Test multiple_meas with no airline""" print(sys._getframe().f_code.co_name) try: dataset_list = pc.multiple_meas(file_path=self.file_path) self.assertIsNotNone(dataset_list) assert len(dataset_list) == 2 except Exception as e: raise plt.close('all')
def test_multiple_meas(self): """Test multiple_meas""" print(sys._getframe().f_code.co_name) try: dataset_list = pc.multiple_meas(file_path=self.file_path, airline_name='VAL') self.assertIsNotNone(dataset_list) assert len(dataset_list) == 2 except Exception as e: raise plt.close('all')
def test_multiple_meas_nofile(self): """Test multiple_meas with no file_path given""" print(sys._getframe().f_code.co_name) try: with patch('permittivitycalc.helper_functions._prompt') as mock: mock.return_value = self.file_path file = mock.return_value dataset_list = pc.multiple_meas(airline_name='VAL') self.assertIsNotNone(dataset_list) assert len(dataset_list) == 2 except Exception as e: raise plt.close('all')