baggage_calculator.py

import unittest
import math 

"""
    Norwegian beskriver reglene for bagasje slik:
    * Hvert enkelt kolli må ikke veie mer enn 23 kg 
    eller mindre enn 2 kg.
    * Du må betale overvekt for kolli som veier 
    over 23 kg. Prisen for overvekt er 135 NOK per kilo

    Skriv en kalkulator som beregner hvor mye man må 
    betale for en gitt vekt.
    Input vil være et desimaltall mellom 2 og 23 kilo.
    Prisen er per begynte kilo rundet oppover.
"""
def bagasje_overvekt(v):
    pris = 0
    overvekt = 135
    if v > 23:
        v = math.ceil(v)
        v = v - 23
        pris = v * overvekt
    return pris

class BaggageCalc(unittest.TestCase):
    def test_underweight(self):
        self.assertEquals(bagasje_overvekt(5.5), 0)

    def test_overweight(self):
        self.assertEquals(bagasje_overvekt(30), 945)
    
    def test_rounded(self):
        self.assertEqual(bagasje_overvekt(23.5), 135)

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