forest - не подходящее название для функции. fight - подходящее.
code для корректного отображения.
locators.py, element.py, page.py и в них классы, причем одни модули импортируются в другие, классы наследуются от импортированных классов и всё довольно стройно.
if работает корректно. if отдельно? Должно, по-идее, создать файл, хотя бы пустой. Создает?