def test_parse_interval_months_arg_raises_error_when_arg_is_not_positive( self): """Zero or negative values for time should raise a ValueError.""" with self.assertRaises(ValueError): cli.parse_interval_months_arg('-5') with self.assertRaises(ValueError): cli.parse_interval_months_arg('0')
def test_parse_interval_months_arg_succeeds_with_valid_arg(self): self.assertEqual( relativedelta.relativedelta(months=1), cli.parse_interval_months_arg('1')) self.assertEqual( relativedelta.relativedelta(months=36), cli.parse_interval_months_arg('36'))