def test_iges_plot_failing_outfile_type(self): iges_handler = IgesHandler() with self.assertRaises(TypeError): iges_handler.plot(plot_file=3)
def test_iges_plot_save_fig(self): iges_handler = IgesHandler() mesh_points = iges_handler.parse('tests/test_datasets/test_pipe.iges') iges_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')