class TestTimerange(object):
timerange_samples = [
jt.timerange(date(1, 1, 1), date(1, 1, 1), jt.DAY),
jt.timerange(date(1, 1, 1), date(1, 1, 1), -jt.DAY),
jt.timerange(date(1, 1, 1), date(1, 1, 2), jt.DAY),
jt.timerange(date(1, 1, 2), date(1, 1, 1), -jt.DAY),
jt.timerange(date(1, 1, 1), date(1, 1, 1), jt.DAY),
jt.timerange(date(2, 2, 2), date(1, 1, 1), jt.DAY),
jt.timerange(date(1, 1, 1), date(2, 2, 2), -jt.DAY),
jt.timerange(datetime(1, 1, 1), datetime(1, 1, 1, 1), jt.DAY),
jt.timerange(datetime(1, 1, 2), datetime(1, 1, 1), -jt.DAY),
jt.timerange(datetime(1, 1, 1, 1), datetime(1, 1, 1, 1), jt.DAY),
jt.timerange(datetime(1, 1, 1), datetime(2, 2, 2), -jt.DAY),
jt.timerange(datetime(2, 2, 2), datetime(1, 1, 1), jt.DAY),
]
def test_init_raise_value_error(self):
with pytest.raises(ValueError) as exc_info:
jt.timerange(datetime(2017, 1, 1), datetime(2017, 2, 2),
timedelta())
msg = 'timerange() arg 3 must not be timedelta(0)'
assert exc_info.value.args == (msg, )
@pytest.mark.parametrize('func, arg_pos', [
(lambda: jt.timerange(100, datetime(2017, 2, 3)), 1),
(lambda: jt.timerange(date(2017, 2, 3), 4), 2),
(lambda: jt.timerange(date(2017, 2, 3), date(2017, 4, 5), 86400), 3),
])
def test_init_raise_type_error(self, func, arg_pos):
with pytest.raises(TypeError) as exc_info:
func()
arg_type = 'timedelta' if arg_pos == 3 else 'date or datetime'
assert exc_info.value.args == (
'timerange() arg {} must be {} object'.format(arg_pos, arg_type), )
@pytest.mark.parametrize('start, stop, step, results', [
(date(1970, 1, 1), date(1970, 2, 1), 15 * jt.DAY,
[date(1970, 1, 1),
date(1970, 1, 16),
date(1970, 1, 31)]),
(date(1, 1, 1), date(2, 2, 2), 365 * jt.DAY,
[date(1, 1, 1), date(2, 1, 1)]),
(date(2, 2, 2), date(1, 1, 1), -365 * jt.DAY,
[date(2, 2, 2), date(1, 2, 2)]),
(date(1, 1, 1), date(1, 1, 1), jt.DAY, []),
(date(1, 1, 1), date(1, 1, 1), -jt.DAY, []),
])
def test_iter_date(self, start, stop, step, results):
assert list(jt.timerange(start, stop, step)) == results
@pytest.mark.parametrize('start, stop, step, results', [
(datetime(1970, 1, 1), datetime(1970, 1, 1, 23), 10 * jt.HOUR, [
datetime(1970, 1, 1),
datetime(1970, 1, 1, 10),
datetime(1970, 1, 1, 20)
]),
(datetime(1, 1, 1, 1), datetime(2, 2, 2, 2), 365 * jt.DAY,
[datetime(1, 1, 1, 1), datetime(2, 1, 1, 1)]),
(datetime(2, 2, 2, 2), datetime(1, 1, 1, 1), -365 * jt.DAY,
[datetime(2, 2, 2, 2), datetime(1, 2, 2, 2)]),
(datetime(2, 2, 2, 2), datetime(2, 2, 2, 2), jt.MICROSECOND, []),
(datetime(2, 2, 2, 2), datetime(2, 2, 2, 2), -jt.MICROSECOND, [])
])
def test_iter_datetime(self, start, stop, step, results):
assert list(jt.timerange(start, stop, step)) == results
@pytest.mark.parametrize('timerange_object', timerange_samples)
def test_bool(self, timerange_object):
assert bool(timerange_object) == bool(list(timerange_object))
@pytest.mark.parametrize('timerange_object', timerange_samples)
def test_timerange_length(self, timerange_object):
assert len(timerange_object) == len(list(timerange_object))
@pytest.mark.parametrize('timerange_object', timerange_samples)
def test_to_str_and_repr(self, timerange_object):
answer = "timerange(start={}, stop={}, step={})".format(
repr(timerange_object.start),
repr(timerange_object.stop),
repr(timerange_object.step),
)
assert str(timerange_object) == repr(timerange_object) == answer
def test_equal(self):
assert jt.timerange(date(1, 1, 1),
date(1, 1,
1)) == jt.timerange(date(1, 1, 1),
date(1, 1, 1))
def test_not_equal(self):
assert jt.timerange(date(1, 1, 1), date(1, 1, 1)) != jt.timerange(
date(1, 1, 2), date(1, 1, 1))
def test_start_not_equal(self):
assert jt.timerange(date(1, 1, 1), date(1, 1, 1)) != jt.timerange(
date(1, 1, 2), date(1, 1, 1))
def test_stop_not_equal(self):
assert jt.timerange(date(1, 1, 2), date(1, 1, 1)) != jt.timerange(
date(1, 1, 2), date(1, 1, 2))
def test_step_not_equal(self):
assert jt.timerange(date(1, 1, 2), date(1, 1, 2),
jt.DAY) != jt.timerange(date(1, 1, 2), date(
1, 1, 2), jt.MINUTE)
def test_type_not_equal(self):
assert jt.timerange(datetime(1, 1, 1), datetime(1, 1,
1)) != timedelta(1)
def test_hash(self):
assert hash(jt.timerange(date(1, 1, 1), date(1, 1, 1))) == hash(
jt.timerange(date(1, 1, 1), date(1, 1, 1)))
def test_not_equal(self):
assert jt.timerange(date(1, 1, 1), date(1, 1, 1)) != jt.timerange(
date(1, 1, 2), date(1, 1, 1))
def test_iter_datetime(self, start, stop, step, results):
assert list(jt.timerange(start, stop, step)) == results
def test_init_raise_value_error(self):
with pytest.raises(ValueError) as exc_info:
jt.timerange(datetime(2017, 1, 1), datetime(2017, 2, 2),
timedelta())
msg = 'timerange() arg 3 must not be timedelta(0)'
assert exc_info.value.args == (msg, )
def test_hash(self):
assert hash(jt.timerange(date(1, 1, 1), date(1, 1, 1))) == hash(
jt.timerange(date(1, 1, 1), date(1, 1, 1)))
def test_type_not_equal(self):
assert jt.timerange(datetime(1, 1, 1), datetime(1, 1,
1)) != timedelta(1)
def test_step_not_equal(self):
assert jt.timerange(date(1, 1, 2), date(1, 1, 2),
jt.DAY) != jt.timerange(date(1, 1, 2), date(
1, 1, 2), jt.MINUTE)
