Python needs_plots Examples

Example #1

 def test_needs_plots(self):
     command_list = ['set_xlim([1, 2])']
     self.assertTrue(needs_plots(command_list))

Example #2

 def test_doesnt_need_plots(self):
     command_list = ['df.reset_index()', 'df.head()']
     self.assertFalse(needs_plots(command_list))