"""
Test case generator for integer literals in the Stack Language.
Generates comprehensive test cases for all integer types and bases.
"""

from .integer_tests import generate_integer_literal_tests
from .float_tests import generate_float_literal_tests

if __name__ == "__main__":
    # import json
    import yaml
    
    # Generate tests
    tests = []
    tests += generate_integer_literal_tests()
    tests += generate_float_literal_tests()
    
    # Print summary
    print(f"Generated {len(tests)} test cases")
    
    # # Save as JSON
    # with open("integer_literal_tests.json", "w") as f:
    #     json.dump(tests, f, indent=2)
    # print("Saved to integer_literal_tests.json")
    
    # # Save as YAML
    # with open("integer_literal_tests.yaml", "w") as f:
    #     yaml.dump(tests, f, default_flow_style=False, sort_keys=False)
    # print("Saved to integer_literal_tests.yaml")
    
    # Save as YAML
    with open("cases.yaml", "w") as f:
        yaml.dump(tests, f, default_flow_style=False, sort_keys=False)
    
    # Print first few tests as example
    # print("\nFirst 3 test cases:")
    # print(yaml.dump(tests[:3], default_flow_style=False, sort_keys=False))