def test_create_continuous_gantt_chart(self): solution_obj = SolutionFactory(csv_data).get_solution() # test without suffix solution_obj.create_gantt_chart_html_file(self.output_file, continuous=True) self.assertTrue(self.output_file.with_suffix('.html').exists()) output_file = self.output_file.with_suffix('.html') # test with suffix output_file.unlink() solution_obj.create_gantt_chart_html_file(output_file, continuous=True) self.assertTrue(output_file.exists()) # test as string output_file.unlink() solution_obj.create_gantt_chart_html_file(str(output_file), continuous=True) self.assertTrue(output_file.exists())