PC-HELP.CZ

Ahoj,
chtěl jsem se optat , v učebnici mají kod, kod který se má otestovat, tak v něm mají přímo assert, označil jsem tučně, zde je chyba, že? Assert se dava jen v testu kodu. Chápu to dobře.
A ještě jeden dotaz, jak propojim test a soubor s testovaným kodem, nějakým IMPORT a <nazev souboru> ??
Dekuji

kod:
KOD K OTESTOVANI:
def get_short_class_description(class_name):
assert class_name == "warrior" #CHYBA, ZDE BY ASSERT NEMELO BYT ASI
return "A battle-hardened veteran, favors heavy armor and weapons."

TEST:
def test_warrior_short_description():
desc = get_short_class_description("warrior")
assert (
desc
== "A battle-hardened veteran, can equip heavy armor and weapons."
)

Ve zdrojaku aplikace pouzivej if/else.
V testu budes mit import short_class_description from abc.def, tady zalezi kde presne se ta funkce nachazi (v jakem souboru, pripadne zanoreni slozek).

dekuji moc

Dodatečně přidáno po 1 hodině 54 minutách 24 vteřinách:
ja se Lama, nejde mi to, mohl bz mi nekdo napsat zhruba, jak ma vypadat IMPORT zapis to test_ souboru.

Import def get_classes_starting_health() from demo2.py

takhle mi to nejde. Demo2.py je soubor s funkci def get_classes_starting_health() , kterou potrebuji priradit do test_ souboru.
Dekuji

From Demo2 import ....<jmeno funkce>

pardon, vždycky to otočím :) to je z toho jak přepínám python a typescript několikrát denně...

from demo2 import get_classes_starting_health