| """Runs example requests through Odyssey_PyPro_10M, prints the generated code, and executes it on a sample list. |
| python examples.py |
| """ |
| import os |
| import sys |
|
|
| sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) |
| from odyssey_pypro import REJECT, generate, load |
|
|
| EXAMPLES = [ |
| ('Write a function named "top" with parameter "xs". Keep numbers at least 3, then sort them from largest to smallest.', |
| [5, 1, 3, 9, 2, 3]), |
| ('Create a function called "clean" that takes a list "vals". Remove duplicate numbers, then return how many numbers there are.', |
| [4, 4, 2, 7, 2]), |
| ('Given a list called "data", write a function named "shift" that does this: Add -2 to each number.', [1, 2, 3]), |
| ('Square each number, then return the sum of the numbers. Name the function "sq_sum" and its parameter "nums".', [1, -2, 3]), |
| ('Using the input list "seq", implement the function "head3". Take the first 3 values.', [9, 8, 7, 6, 5]), |
| ('Define "low", a function whose argument is "arr". Keep only values below 0, then find the minimum value.', [3, -4, 0, -1]), |
| ('Write a function named "med" with parameter "xs". Compute the median of the numbers.', [1, 2, 3]), |
| ] |
|
|
| if __name__ == "__main__": |
| model = load() |
| for text, sample in EXAMPLES: |
| src = generate(model, text) |
| print("REQUEST:", text) |
| print(src, end="") |
| if src != REJECT: |
| ns = {} |
| exec(src, ns) |
| fn = [v for k, v in ns.items() if callable(v) and not k.startswith("__")][0] |
| print(f"# {fn.__name__}({sample}) -> {fn(list(sample))}") |
| print() |
|
|
