from fag import Fag
from student import Student

def hovedprogram():
    in1000 = Fag("IN1000: Introduksjon til objektorientert programmering")
    in1020 = Fag("IN1020: Introduksjon til datateknologi")
    
    per = Student("Per")

    in1000.legg_til_student(per)
    per.legg_til_fag(in1000)

    in1020.legg_til_student(per)
    per.legg_til_fag(in1020)

    per.skriv_fag_paa_student()

    print()

    anne = Student("Anne")
    in1000.legg_til_student(anne)
    anne.legg_til_fag(in1000)

    anne.skriv_fag_paa_student()

    print()

    in1000.skriv_studenter_ved_fag()

    print("Antall studenter:", in1000.hent_antall_studenter())

    assert in1000.hent_antall_studenter() == 2

    assert in1020.hent_antall_studenter() == 1

hovedprogram()