DeclensionTest.py

import unittest
from Declension import Declension


class DeclensionTest(unittest.TestCase):

    def test_1(self):
        Declension.set("usd", ["доллар", "доллара", "долларов"])
        data = Declension.get(1, "usd")
        self.assertEqual("доллар", data)

    # noinspection SpellCheckingInspection
    def test_2(self):
        Declension.set("грн", ["гривня", "гривні", "гривень"])
        data = Declension.format("2", "грн")
        self.assertEqual("2 гривні", data)

    def test_3(self):
        Declension.set("руб", ["рубль", "рубля", "рублей"])
        mask = "На Вашем счету {item} {form}!"
        data = Declension.format("5.25", "руб", mask)
        self.assertEqual("На Вашем счету 5 рублей!", data)

    def test_4(self):
        Declension.set("руб", ["рубль", "рубля", "рублей"])
        mask = "На Вашем счету {item} {form}!"
        data = Declension.format(-5.25, "руб", mask)
        self.assertEqual("На Вашем счету -6 рублей!", data)

    def test_5(self):
        Declension.set("дн.", ["день", "дня", "дней"])
        mask = "Интернет отключится через {item} {form}"
        data = Declension.format(22, "дн.", mask)
        self.assertEqual("Интернет отключится через 22 дня", data)


# Точка входа
if __name__ == "__main__":
    unittest.main()