executor/src/test/resources/runtime.py

class AmaContext(object):

    def __init__(self, sc, sqlContext):
        self.sc = sc
        self.sqlContext = sqlContext