test_quick.py

"""
Quick test to verify the materials.py file can be executed.
"""

import sys
import os

def test_import():
    """Test that materials.py can be imported and executed."""
    print("Testing materials.py execution...")
    
    # Read and execute the materials file
    materials_path = os.path.join("benchmark", "application", "materials.py")
    
    if not os.path.exists(materials_path):
        print(f"ERROR: {materials_path} not found")
        return False
    
    with open(materials_path, "r", encoding="utf-8") as f:
        code = f.read()
    
    # Execute in a clean namespace
    exec_globals = {}
    try:
        exec(code, exec_globals)
        print("SUCCESS: materials.py executed successfully")
        
        # Check for required variables
        required_vars = ['m', 'x', 'material_quality', 'material_abundance', 
                        'material_usage', 'material_cost']
        
        missing = []
        for var in required_vars:
            if var not in exec_globals:
                missing.append(var)
        
        if missing:
            print(f"ERROR: Missing variables: {', '.join(missing)}")
            return False
        
        print("SUCCESS: All required variables found")
        
        # Check model status
        model = exec_globals['m']
        if hasattr(model, 'status'):
            if model.status == 2:  # GRB.OPTIMAL
                print(f"SUCCESS: Model solved to optimality (cost: {model.objVal:.4f})")
            else:
                print(f"WARNING: Model status: {model.status} (not optimal)")
        
        return True
        
    except ImportError as e:
        if 'gurobipy' in str(e):
            print("ERROR: Gurobi not installed. Please install gurobipy:")
            print("  pip install gurobipy")
            print("  (Also requires Gurobi Optimizer license)")
        else:
            print(f"ERROR: Import error: {e}")
        return False
    except Exception as e:
        print(f"ERROR: {e}")
        import traceback
        traceback.print_exc()
        return False


if __name__ == "__main__":
    success = test_import()
    sys.exit(0 if success else 1)