def test_draw_scree_graph(self): """draw_scree_graph: draws the matplotlib figure""" filename1 = join(self.tmp_dir,'scree_plot.png') filename2 = join(self.tmp_dir,'scree_plot.eps.gz') self._paths_to_clean_up = [filename1,filename2] obs1,obs2=draw_scree_graph(self.tmp_dir, self.tmp_dir, self.background_color, self.label_color, generate_eps = True, data = self.data) self.assertEqual(obs1,expimgsrc_scree % filename1) self.assertEqual(obs2,expdownlink_scree % filename2) self.assertTrue(exists(filename1),'The png file was not created in the appropriate location') self.assertTrue(exists(filename2),'The eps file was not created in the appropriate location')