from unittest import TestCase
from osztondij_kalkulator import osztondijErtek

class Test(TestCase):
    def test_osztondij0(self):
        self.assertEqual(osztondijErtek("I", 3, 12, 4.2), 0)
        self.assertEqual(osztondijErtek("N", 7, 10, 3.67), 0)
        self.assertEqual(osztondijErtek("N", 0, 11, 1.9), 0)
    def test_osztondij8000(self):
        self.assertEqual(osztondijErtek("N", 1, 9, 4.2), 8000)
        self.assertEqual(osztondijErtek("N", 0, 10, 2.5), 8000)
        self.assertEqual(osztondijErtek("N", 0, 11, 2.07), 8000)
    def test_osztondij25000(self):
        self.assertEqual(osztondijErtek("N", 0, 13, 3.21), 25000)
    def test_osztondij42000(self):
        self.assertEqual(osztondijErtek("N", 0, 13, 4.00), 42000)
    def test_osztondij59000(self):
        self.assertEqual(osztondijErtek("N", 0, 12, 4.72), 59000)