test_parser.py - master - dayekb/vibecoding_vkr_citation_parser

test_parser.py
72 строки3 KB
dayekb
upload files
05 окт 2025, 14:32
05 окт 2025, 14:32d089abf
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import os
import sys
from citation_parser import CitationParser

def test_parser():
    """Тестирование парсера на дипломных файлах"""

    # Путь к дипломным файлам
    docx_file = "Диплом_Гайнутдинов_Тимур.docx"
    pdf_file = "Диплом_Гайнутдинов_Тимур.pdf"

    print("Тестирование парсера цитат дипломной работы")
    print("=" * 50)

    # Тестируем DOCX файл
    if os.path.exists(docx_file):
        print(f"\nТестирование DOCX файла: {docx_file}")
        try:
            parser = CitationParser(docx_file)
            result = parser.parse()

            print(f"Найдено ссылок в списке литературы: {len(result['references'])}")
            print(f"Найдено цитат в тексте: {len(result['citations'])}")

            # Выводим первые несколько цитат для примера
            print("\nПервые 5 цитат:")
            for i, citation in enumerate(result['citations'][:5]):
                print(f"{i+1}. Цитата №{citation['citation_number']}:")
                print(f"   Контекст: {citation['context'][:100]}...")
                print(f"   Полная ссылка: {citation['citation'][:100]}...")
                print()

        except Exception as e:
            print(f"Ошибка при обработке DOCX файла: {e}")

    # Тестируем PDF файл
    if os.path.exists(pdf_file):
        print(f"\nТестирование PDF файла: {pdf_file}")
        try:
            parser = CitationParser(pdf_file)
            result = parser.parse()

            print(f"Найдено ссылок в списке литературы: {len(result['references'])}")
            print(f"Найдено цитат в тексте: {len(result['citations'])}")

            # Выводим первые несколько цитат для примера
            print("\nПервые 5 цитат:")
            for i, citation in enumerate(result['citations'][:5]):
                print(f"{i+1}. Цитата №{citation['citation_number']}:")
                print(f"   Контекст: {citation['context'][:100]}...")
                print(f"   Полная ссылка: {citation['citation'][:100]}...")
                print()

        except Exception as e:
            print(f"Ошибка при обработке PDF файла: {e}")

    # Сохранение результатов в JSON файлы
    print("\nСохранение результатов в JSON файлы:")
    if os.path.exists(docx_file):
        parser = CitationParser(docx_file)
        parser.save_to_json()

    if os.path.exists(pdf_file):
        parser = CitationParser(pdf_file)
        parser.save_to_json()


if __name__ == "__main__":
    test_parser()