Atomizer/optimization_engine/hooks/test_introspection.py

"""Test model introspection module."""

import json
import glob
from pathlib import Path

# Add project root to path
import sys
sys.path.insert(0, str(Path(__file__).parent.parent.parent))

from optimization_engine.hooks.nx_cad.model_introspection import (
    introspect_op2,
    introspect_study,
)


def test_op2_introspection():
    """Test OP2 introspection on bracket study."""
    print("=" * 60)
    print("OP2 INTROSPECTION TEST")
    print("=" * 60)

    # Find bracket OP2 files
    op2_files = glob.glob(
        "C:/Users/Antoine/Atomizer/studies/bracket_stiffness_optimization_V3/**/*.op2",
        recursive=True
    )

    print(f"\nFound {len(op2_files)} OP2 files")
    for f in op2_files[:5]:
        print(f"  - {Path(f).name}")

    if not op2_files:
        print("No OP2 files found!")
        return

    # Introspect first OP2
    print(f"\nIntrospecting: {Path(op2_files[0]).name}")
    result = introspect_op2(op2_files[0])

    if not result["success"]:
        print(f"ERROR: {result['error']}")
        return

    data = result["data"]

    # Print results
    print(f"\nFile Info:")
    print(f"  Size: {data['file_info']['size_mb']:.2f} MB")
    print(f"  Subcases: {data['subcases']}")

    print(f"\nAvailable Results:")
    for r_type, info in data["results"].items():
        status = "YES" if info["available"] else "no"
        extra = ""
        if info["available"]:
            if "element_types" in info and info["element_types"]:
                extra = f" ({', '.join(info['element_types'][:3])})"
            elif "subcases" in info and info["subcases"]:
                extra = f" (subcases: {info['subcases'][:3]})"
        print(f"  {r_type:20s}: {status:4s} {extra}")

    print(f"\nMesh Info:")
    print(f"  Nodes: {data['mesh']['node_count']}")
    print(f"  Elements: {data['mesh']['element_count']}")
    if data['mesh']['element_types']:
        print(f"  Element types: {list(data['mesh']['element_types'].keys())[:5]}")

    print(f"\nExtractable results: {data['extractable']}")


def test_study_introspection():
    """Test study directory introspection."""
    print("\n" + "=" * 60)
    print("STUDY INTROSPECTION TEST")
    print("=" * 60)

    study_dir = "C:/Users/Antoine/Atomizer/studies/bracket_stiffness_optimization_V3"
    print(f"\nIntrospecting study: {study_dir}")

    result = introspect_study(study_dir)

    if not result["success"]:
        print(f"ERROR: {result['error']}")
        return

    data = result["data"]

    print(f"\nStudy Summary:")
    print(f"  Parts (.prt): {data['summary']['part_count']}")
    print(f"  Simulations (.sim): {data['summary']['simulation_count']}")
    print(f"  Results (.op2): {data['summary']['results_count']}")
    print(f"  Has config: {data['summary']['has_config']}")

    print(f"\nParts found:")
    for p in data["parts"][:5]:
        print(f"  - {Path(p['path']).name}")

    print(f"\nSimulations found:")
    for s in data["simulations"][:5]:
        print(f"  - {Path(s['path']).name}")

    if data["config"]:
        print(f"\nOptimization Config:")
        config = data["config"]
        if "variables" in config:
            print(f"  Variables: {len(config['variables'])}")
            for v in config["variables"][:3]:
                print(f"    - {v.get('name', 'unnamed')}: [{v.get('lower')}, {v.get('upper')}]")
        if "objectives" in config:
            print(f"  Objectives: {len(config['objectives'])}")
            for o in config["objectives"][:3]:
                print(f"    - {o.get('name', 'unnamed')} ({o.get('direction', 'minimize')})")


if __name__ == "__main__":
    test_op2_introspection()
    test_study_introspection()
    print("\n" + "=" * 60)
    print("INTROSPECTION TESTS COMPLETE")
    print("=" * 60)