#!/usr/bin/env python3

try:
    from iotest import run_assignment_tests
except ImportError:
    print("❌ Error: Could not find the 'iotest' module.")
    print("Please make sure 'iotest.py' is in the same directory as this test file and your solution program.")
    exit(1)

test_cases = [
    {"args": ["3", "2"], "expected_output": "12"},
    {"args": ["1", "9"], "expected_output": "9"},
    {"args": ["10", "2"], "expected_output": "110"},
    {"args": ["10", "5"], "expected_output": "275"},
    {"args": ["4", "141"], "expected_output": "1410"},
    {"args": ["98", "7"], "expected_output": "33957"},
    {"args": ["100", "1"], "expected_output": "5050"},
]

if __name__ == "__main__":
    run_assignment_tests("p2.py", test_cases)