Spaces:
Sleeping
Sleeping
| import sys, io, os | |
| sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8') | |
| sys.path.insert(0, '.') | |
| from pipeline.resolver import resolve | |
| from pipeline.fetcher import fetch_all | |
| from pipeline.calculator import calculate | |
| from pipeline.analyst import generate_narrative | |
| from builders.build_excel import build_workbook | |
| from builders.build_docx import build_document | |
| def run_test(): | |
| print("1. Resolving TCS...") | |
| meta = resolve('TCS', log_fn=lambda m: print(" " + m)) | |
| print("\n2. Fetching data...") | |
| data = fetch_all(meta, years=5, force_refresh=False, log_fn=lambda m: print(" " + m)) | |
| print("\n3. Calculating ratios...") | |
| data = calculate(data, meta.to_dict()) | |
| print("\n4. Generating Narrative (Data-only)...") | |
| # Use empty models list to skip actual API calls for a quick formatting test | |
| narrative = generate_narrative(data, meta.to_dict(), models=[], log_fn=lambda m: print(" " + m)) | |
| data.update(narrative) | |
| # Save test files | |
| os.makedirs("test_outputs", exist_ok=True) | |
| excel_path = "test_outputs/TCS_Analysis.xlsx" | |
| docx_path = "test_outputs/TCS_Analysis.docx" | |
| print("\n5. Building Excel...") | |
| build_workbook(data, excel_path) | |
| print(f" ✓ Saved to {excel_path}") | |
| print("\n6. Building Word Document...") | |
| build_document(data, docx_path) | |
| print(f" ✓ Saved to {docx_path}") | |
| print("\nAll tests passed successfully!") | |
| if __name__ == "__main__": | |
| run_test() | |