def assert_eq(expr, expected=None):
sql = FugueSQL(expr, "fugueWildSchema", ignore_case=True)
v = _VisitorBase(sql)
obj = v.visit(sql.tree)
if expected is None:
expected = expr
assert expected == obj
def assert_eq(expr, varname, sign):
sql = FugueSQL(expr,
"fugueAssignment",
ignore_case=True,
simple_assign=True)
v = _VisitorBase(sql)
obj = v.visit(sql.tree)
assert (varname, sign) == obj
def assert_eq(expr, expected):
sql = FugueSQL(expr, "fugueJsonValue", ignore_case=True)
v = _VisitorBase(sql)
obj = v.visit(sql.tree)
if expected is None:
assert obj is None
else:
assert expected == obj
def assert_eq(expr, is_checkpoint, value):
sql = FugueSQL(expr,
"fuguePersist",
ignore_case=True,
simple_assign=True)
v = _VisitorBase(sql)
obj = v.visit(sql.tree)
assert (is_checkpoint, value) == obj
def assert_eq(expr, using, params, schema):
sql = FugueSQL(expr,
"fugueSingleOutputExtensionCommon",
ignore_case=True)
v = _VisitorBase(sql)
obj = v.visit(sql.tree)
assert using == obj["using"]
if params is None:
assert "params" not in obj
else:
assert params == obj["params"]
if schema is None:
assert "schema" not in obj
else:
assert obj["schema"] == schema
def assert_eq(expr, expected):
sql = FugueSQL(expr, "fuguePrepartition", ignore_case=True)
v = _VisitorBase(sql)
obj = json.dumps(v.visit(sql.tree).jsondict)
assert json.dumps(expected.jsondict) == obj
def assert_eq(expr, expected):
sql = FugueSQL(expr, "fugueParams", ignore_case=True)
v = _VisitorBase(sql)
obj = v.visit(sql.tree)
assert expected == obj
