from fag import Fag
from student import Student

def hovedprogram():
    in1000 = Fag("IN1000")
    in1020 = Fag("IN1020")
    kari = Student("Kari")
    ole = Student("Ole")

    # Må kalle på begge metodene (?) for å registrere student på fag
    ole.leggTilFag(in1000)
    in1000.leggTilStudent(ole)

    # Eventuelt lage en prosedyre for å forenkle dette.
    registrer(in1020, ole)
    registrer(in1000, kari)

    # Tester metoder for navn og antall:
    print(in1000.hentFagNavn(), "har", in1000.hentAntallStudenter(), "studenter.")
    print(kari.hentStudentNavn(), "tar", kari.hentAntallFag(), "fag.")

    # Tester med assert
    antStud = in1000.hentAntallStudenter()
    assert antStud == 2, "in1000.hentAntallStudenter() == " + str(antStud)

    antFag = kari.hentAntallFag()
    assert antFag == 1, "kari.hentAntallFag() == " + str(antFag)

    # Tester utskriftene
    in1000.skrivStudenterVedFag()
    ole.skrivFagPaaStudent()



def registrer(fag, student):
    fag.leggTilStudent(student)
    student.leggTilFag(fag)

hovedprogram()