def _get_room_filter(cls, number):
if number == 1:
number_word = number_to_text(number)
suffix = 'dormitorio'
elif number >= 4:
four_str = number_to_text(4)
five_str = number_to_text(5)
number_word = f'{four_str}-{five_str}'
suffix = 'habitaciones-o-mas'
else:
number_word = number_to_text(number)
suffix = 'dormitorios'
return f'de-{number_word}-{suffix}'
def test_number_zero(self):
number_zero = number_to_text(0)
assert 'cero' == number_zero
def test_number_greater_than_max_number(self):
with pytest.raises(ValueError):
number_to_text(MAX_NUMBER + 1)
def test_max_number(self):
number_to_text(MAX_NUMBER)
assert True # if no number_to_text fails, `assert True` will be never achieved
def test_number_nine(self):
number_nine = number_to_text(9)
assert 'nueve' == number_nine
def test_number_one(self):
number_one = number_to_text(1)
assert 'un' == number_one