class TestWriteOpener(unittest.TestCase): def setUp(self): self.write_opener = WriteOpener() @mock.patch('tests.files.openers.test_write_opener.builtins.open') def test_open_file_write_byte_strings(self, mock_open_function): mock_filename = "file.csv" mock_file = mock.Mock() mock_open_function.return_value = mock_file result = self.write_opener.open_file_write_byte_strings(mock_filename) self.assertEquals(result, mock_file) @mock.patch('tests.files.openers.test_write_opener.builtins.open') def test_open_file_write_unicode_strings(self, mock_open_function): mock_filename = "file.csv" mock_file = mock.Mock() mock_open_function.return_value = mock_file result = self.write_opener.open_file_write_unicode_strings(mock_filename) self.assertEquals(result, mock_file)
def setUp(self): self.write_opener = WriteOpener()