def test_step_plot_failing_outfile_type(self):
step_handler = StepHandler()
with self.assertRaises(TypeError):
step_handler.plot(plot_file=3)
def test_step_plot_save_fig(self):
step_handler = StepHandler()
mesh_points = step_handler.parse('tests/test_datasets/test_pipe.step')
step_handler.plot(save_fig=True)
self.assertTrue(os.path.isfile('tests/test_datasets/test_pipe.png'))
self.addCleanup(os.remove, 'tests/test_datasets/test_pipe.png')