"""
Корректные примеры:
SALE-1234
SALE-0001
SALE-9999

Некорректные:
sale-1234
SALE1234
SALE-12
SALE-12345
DISCOUNT-1234

"""

import unittest

from main import is_valid_coupon


class CouponTest(unittest.TestCase):
    def test_stadard(self):
        self.assertEqual(True, is_valid_coupon(code="SALE-1234"))
        self.assertEqual(True, is_valid_coupon(code="SALE-0001"))
        self.assertEqual(True, is_valid_coupon(code="SALE-9999"))

    def test_non_correct(self):
        self.assertEqual(False, is_valid_coupon(code="sale-1234"))
        self.assertEqual(False, is_valid_coupon(code="SALE1234"))
        self.assertEqual(False, is_valid_coupon(code="SALE-12"))
        self.assertEqual(False, is_valid_coupon(code="SALE-12345"))
        self.assertEqual(False, is_valid_coupon(code="DISCOUNT-1234"))


if __name__ == "__main__":
    unittest.main()