def test_ensure_xy_args_good(self, exit_mock, stderr_mock): """ensure_xy_args() doesn't exit when args okay """ stderr_mock.write = MagicMock() args = MagicMock(x=None, y=None) plot_lib.ensure_xy_args(args) self.assertFalse(exit_mock.called)
def test_ensure_xy_args_bad(self, exit_mock, stderr_mock): """ensure_xy_args() exits when args are bad """ stderr_mock.write = MagicMock() args = MagicMock(x=None, y=True) plot_lib.ensure_xy_args(args) self.assertTrue(exit_mock.called)