def test_load_pickle_file(self): # Arrange test_file_name = "testdata\\test_pickle_load.txt" expected_data_from_file =\ ['€\x03X\x1d\x00\x00\x00This is being saved to a fileq\x00.'] result = "" # Act result = FileReader.load_pickle_file(FileReader, test_file_name) # Assert self.assertTrue(result == expected_data_from_file)
def test_load_pickle_file_by_inputting_file_name(self): # Arrange user_input = "P" expected_data_from_file =\ ['€\x03X\x1d\x00\x00\x00This is being saved to a fileq\x00.'] result = "" # Act with patch('builtins.input', side_effect=user_input): result = FileReader.load_pickle_file(FileReader) # Assert self.assertTrue(result == expected_data_from_file)
def test_load_pickle_file_error_by_inputting_wrong_file_name(self): # Arrange user_input = "A" expected_string = "File not found" result = "" # Act captured_output = io.StringIO() sys.stdout = captured_output with patch('builtins.input', side_effect=user_input): result = FileReader.load_pickle_file(FileReader) sys.stdout = sys.__stdout__ # Assert self.assertTrue(expected_string in captured_output.getvalue())