def test_date(self):
sequence = seq(
datetime.date(2021, 2, 11), increment_by=datetime.timedelta(days=6)
)
assert next(sequence) == datetime.date(2021, 2, 17)
assert next(sequence) == datetime.date(2021, 2, 23)
assert next(sequence) == datetime.date(2021, 3, 1)
def test_numbers_with_suffix(self):
with pytest.raises(TypeError) as exc:
next(seq(1, suffix="iamnotanumber"))
assert (
str(exc.value) == "Sequences with suffix can only be used with text values"
)
def test_should_raise_exception_for_datetime_instances(self, value):
with pytest.raises(TypeError) as exc:
next(seq(value))
assert str(exc.value) == (
"Sequences with values datetime.datetime, datetime.date and datetime.time, "
"incremente_by must be a datetime.timedelta."
)
def test_time(self):
sequence = seq(
datetime.time(15, 39, 58, 457698),
increment_by=datetime.timedelta(minutes=59),
)
assert next(sequence) == datetime.time(16, 38, 58, 457698)
assert next(sequence) == datetime.time(17, 37, 58, 457698)
assert next(sequence) == datetime.time(18, 36, 58, 457698)
def test_datetime(self, settings, use_tz):
settings.USE_TZ = use_tz
tzinfo = utc if use_tz else None
sequence = seq(
datetime.datetime(2021, 2, 11, 15, 39, 58, 457698),
increment_by=datetime.timedelta(hours=3),
)
assert next(sequence) == datetime.datetime(
2021, 2, 11, 18, 39, 58, 457698).replace(tzinfo=tzinfo)
assert next(sequence) == datetime.datetime(
2021, 2, 11, 21, 39, 58, 457698).replace(tzinfo=tzinfo)
assert next(sequence) == datetime.datetime(
2021, 2, 12, 00, 39, 58, 457698).replace(tzinfo=tzinfo)
def test_decimal(self):
sequence = seq(Decimal("36.6"))
assert next(sequence) == Decimal("37.6")
assert next(sequence) == Decimal("38.6")
assert next(sequence) == Decimal("39.6")
def test_int_increment_by(self):
sequence = seq(1, increment_by=3)
assert next(sequence) == 4
assert next(sequence) == 7
assert next(sequence) == 10
def test_int(self):
sequence = seq(1)
assert next(sequence) == 2
assert next(sequence) == 3
assert next(sequence) == 4
def test_string_invalid_suffix(self):
with pytest.raises(TypeError) as exc:
next(seq("cookie", suffix=42))
assert str(exc.value) == "Sequences suffix can only be a string"
def test_string_suffix_and_start(self):
sequence = seq("cookie", start=111, suffix="@example.com")
assert next(sequence) == "*****@*****.**"
assert next(sequence) == "*****@*****.**"
assert next(sequence) == "*****@*****.**"
def test_string_suffix(self):
sequence = seq("cookie", suffix="@example.com")
assert next(sequence) == "*****@*****.**"
assert next(sequence) == "*****@*****.**"
assert next(sequence) == "*****@*****.**"
def test_string_start(self):
sequence = seq("muffin", start=9)
assert next(sequence) == "muffin9"
assert next(sequence) == "muffin10"
assert next(sequence) == "muffin11"
def test_string(self):
sequence = seq("muffin")
assert next(sequence) == "muffin1"
assert next(sequence) == "muffin2"
assert next(sequence) == "muffin3"
def test_float_increment_by(self):
sequence = seq(1.23, increment_by=1.8)
assert next(sequence) == pytest.approx(3.03)
assert next(sequence) == pytest.approx(4.83)
assert next(sequence) == pytest.approx(6.63)
def test_float(self):
sequence = seq(1.23)
assert next(sequence) == 2.23
assert next(sequence) == 3.23
assert next(sequence) == 4.23
def test_decimal_increment_by(self):
sequence = seq(Decimal("36.6"), increment_by=Decimal("2.4"))
assert next(sequence) == Decimal("39.0")
assert next(sequence) == Decimal("41.4")
assert next(sequence) == Decimal("43.8")
