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