def test_bad_cols(self, stderr_mock): # passing test means nothing raised args = MagicMock(quiet=False) cols = ['b'] df = MagicMock(columns=['a']) with self.assertRaises(SystemExit): validate_args(args, cols, df)
def test_okay(self): # passing test means nothing raised args = MagicMock(quiet=False) cols = ['a'] df = MagicMock(columns=['a']) validate_args(args, cols, df)