def test_csv_with_broken_file(self, mock_dataframe, mock_config_locations,
mock_parse):
self.tearDown()
mock_config_locations.return_value = [self._path]
mock_dataframe.side_effect = DataProviders.get_csv_results()
with patch('pyccata.core.managers.clients.docx.Docx') as docx:
docx.__implements__ = (ReportingInterface, )
document = DocumentController('broken_csv.json')
document.build()
csvfiles = document._thread_manager.projectmanager._client._client
self.assertIsInstance(csvfiles, CSVClient)
self.assertEquals(len(csvfiles), 0)
document.save('Test Document.docx')
def test_csv_with_multi_file(self, mock_dataframe, mock_config_locations,
mock_parse):
mock_config_locations.return_value = [self._path]
mock_dataframe.side_effect = DataProviders.get_csv_results()
with patch('pyccata.core.managers.clients.docx.Docx') as docx:
docx.__implements__ = (ReportingInterface, )
document = DocumentController('csv_multi_file.json')
document.build()
self.assertEquals(
None, document._thread_manager.projectmanager._client.server)
self.assertIsInstance(
document._thread_manager.projectmanager._client.projects(),
list)
csvfiles = document._thread_manager.projectmanager._client._client
self.assertIsInstance(csvfiles, CSVClient)
self.assertEquals(len(csvfiles), 2)
for csv in csvfiles:
self.assertIsInstance(csv._dataframe, pandas.DataFrame)
self.assertEquals(len(csv._dataframe.index), 3)
document.save('Test Document.docx')