from services.timetable_service import TimetableService


def test_train_lookup():
    service = TimetableService()

    result = service.get_full_info(
        train_number=1001,
        diagram_type="weekday"
    )

    print("\n=== 열차 정보 ===")
    print(result["meta"])

    print("\n=== 시각표 ===")
    for row in result["timetable"][:5]:
        print(row)


if __name__ == "__main__":
    test_train_lookup()
