27 lines
623 B
Python
Executable File
27 lines
623 B
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import argparse
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Test grader script")
|
|
parser.add_argument("--answer", type=str, required=True, help="Predicted answer")
|
|
parser.add_argument("--expected", type=str, required=True, help="Expected answer")
|
|
args = parser.parse_args()
|
|
|
|
pred = args.answer.strip()
|
|
gold = args.expected.strip()
|
|
|
|
print(f"Gold: {gold}")
|
|
print(f"Pred: {pred}")
|
|
|
|
if pred == gold:
|
|
print("Correct!")
|
|
sys.exit(0)
|
|
else:
|
|
print("Incorrect")
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|