def test_formats():
x = dts.DateTimeSimple()
y = datetime.date.today()
x.set_format('MM/DD/YYYY')
assert str(x) == str(y.strftime('%m/%d/%Y'))
x.set_format('DD-Mon-YY')
assert str(x) == str(y.strftime('%d-%b-%Y'))
x.set_format('YYYY-MM-DD')
assert str(x) == str(y.strftime('%Y-%m-%d'))
def test_bad_set_format():
with pytest.raises(ValueError):
x = dts.DateTimeSimple()
x.set_format('Mon-DD-YYYY')
def test_bad_constructor_format():
with pytest.raises(ValueError):
x = dts.DateTimeSimple('Jun-06-2016')
def test_print_default():
x = dts.DateTimeSimple()
assert str(x) == str(datetime.date.today())
def test_add_days_leap_year():
x = dts.DateTimeSimple('2016-02-28')
x = x + 3
y = datetime.datetime.strptime('2016-03-02', '%Y-%m-%d')
assert str(x) == str(y.strftime('%Y-%m-%d'))
def test_add_two_weeks():
x = dts.DateTimeSimple('2016-06-10')
x.add_week(2)
y = datetime.datetime.strptime('2016-06-24', '%Y-%m-%d')
assert str(x) == str(y.strftime('%Y-%m-%d'))
def test_minus_one_week():
x = dts.DateTimeSimple('2016-06-10')
x.add_week(-1)
y = datetime.datetime.strptime('2016-06-03', '%Y-%m-%d')
assert str(x) == str(y.strftime('%Y-%m-%d'))
def test_minus_one_day():
x = dts.DateTimeSimple('2016-06-06')
x = x - 1
y = datetime.datetime.strptime('2016-06-05', '%Y-%m-%d')
assert str(x) == str(y.strftime('%Y-%m-%d'))