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