from skole import Skole

def main():
    skole = Skole() 
    skole.lesFraFil("studenter.txt")

    elever = skole.hentElever() 
    assert len(elever) == 11 
    skole.leggTilMatteElever()
    matte_elever = skole.hentMatteElever()
    assert len(matte_elever) == 5 

    skole.byttFag()
    skole.leggTilMatteElever()
    matte_elever = skole.hentMatteElever() 
    assert len(matte_elever) == 9

    draHjem = skole.hjemTidlig()
    assert len(draHjem) == 3

main()