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