Example #1
0
def process_answer(answer, extraction = True):
    code = extract_code(answer)
    code = re.sub("\$.*\$", "", code)
    
    try:
        data = pyth.general_parse(code)[1]
    except Exception as e:
        print(code)
        print(e)
        data = {i: [0]*3 for i in "bdGHkNTYZ"}
    
    data = {i: np.array(v + [v[0]+v[2]]) for i, v in data.items()}
    
    data["Code"] = code
    
    return pd.Series(data)
Example #2
0
#!/usr/bin/env python3

import os
import sys
sys.path.append(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'pyth'))
import pyth

import readline
import traceback

while True:
    try:
        exec(pyth.general_parse(input('>>> ')), pyth.environment)
    except EOFError:
        break
    except:
        traceback.print_exc()