from knowledge_base import KnowledgeBase

def test_data():
    print("Тестирование базы знаний...")
    
    kb = KnowledgeBase()
    print(f"Курсов в базе: {len(kb.courses)}")
    
    if kb.courses:
        print("Первые 3 курса:")
        for i, course in enumerate(kb.courses[:3], 1):
            print(f"{i}. {course['name']} (семестр {course.get('semester', 'N/A')})")
    else:
        print("Курсы не найдены!")
    
    print(f"Программ в базе: {len(kb.programs)}")
    
    # Тест поиска по семестрам
    for semester in [1, 2, 3, 4]:
        courses = kb.get_courses_by_semester(semester)
        print(f"Семестр {semester}: {len(courses)} курсов")

if __name__ == '__main__':
    test_data()