示例#1
0
 def test_needs_plots(self):
     command_list = ['set_xlim([1, 2])']
     self.assertTrue(needs_plots(command_list))
示例#2
0
 def test_doesnt_need_plots(self):
     command_list = ['df.reset_index()', 'df.head()']
     self.assertFalse(needs_plots(command_list))