def test_plot_heatmap_does_reset_plot(self, subplot_reset_mock,
plot_reset_mock, sns_reset_mock):
dummy_data = [["Entry name %s" % dummy_val, dummy_val]
for dummy_val in list(range(1, 10))]
dummy_dataframe = pd.DataFrame(dummy_data)
sut.plot_heatmap(dummy_dataframe, "", "", "")
self.assertTrue(sns_reset_mock.called)
self.assertTrue(subplot_reset_mock.called)
self.assertTrue(plot_reset_mock.called)
def test_plot_heatmap_does_plot(self, makedire_mock, heatmap_mock,
close_fig_mock):
dummy_data = [["Entry name %s" % dummy_val, dummy_val]
for dummy_val in list(range(1, 10))]
dummy_dataframe = pd.DataFrame(dummy_data)
sut.plot_heatmap(dummy_dataframe, "", "", "")
self.assertTrue(makedire_mock.called)
self.assertTrue(heatmap_mock.called)
self.assertTrue(close_fig_mock.called)
def test_plot_heatmap_logs_exp_message_for_empty_dataframe(self):
used_file_name = "dummyFileName"
exp_logger_name = sut.__name__
with self.assertLogs(exp_logger_name, level="INFO") as mock_log:
sut.plot_heatmap(pd.DataFrame(), "dummyTitle", "", used_file_name)
self.assertEqual(
mock_log.output,
[
"INFO:" + exp_logger_name +
":No data available. Skip writing heatmap: " +
used_file_name
],
)
def test_plot_heatmap_logs_exp_message_if_dataframe_extends_max_limit(
self):
used_file_name = "dummyFileName"
exp_logger_name = sut.__name__
dummy_data = [["Entry name %s" % dummy_val, dummy_val]
for dummy_val in list(range(0, 201))]
dummy_dataframe = pd.DataFrame(dummy_data)
with self.assertLogs(exp_logger_name, level="INFO") as mock_log:
sut.plot_heatmap(dummy_dataframe, "dummyTitle", "", used_file_name)
self.assertEqual(
mock_log.output,
[
"INFO:" + exp_logger_name +
":Number of entries is 201 and exceeds limit of 200 for heatmaps. Will skip creation of heatmap: "
+ used_file_name
],
)
def write_results(self, output_dir):
self._logger.info("Writing project analysis results")
writer.write_graphml(
os.path.join(output_dir, "project_dependencies.graphml"),
self._project_graph,
)
writer.write_graphml(
os.path.join(output_dir, "cyclic_project_dependencies.graphml"),
self._cycle_project_graph,
)
writer.write_cycles_to_txt(
os.path.join(output_dir, "project_cycles.txt"),
self._project_cycles)
plot.plot_heatmap(
self._project_coupling_data_frame,
"Project Coupling",
output_dir,
"project_coupling_heatmap.pdf",
)
def test_plot_heatmap_does_not_plot_if_dataframe_empty(self, mock_writer):
sut.plot_heatmap(pd.DataFrame(), "", "", "")
self.assertFalse(mock_writer.called)