def test_bad_call_to_adjust_context():
op = "not_a_node"
context = (pd.Timestamp('20170101'), pd.Timestamp('20170103'))
scope = Scope()
with pytest.raises(com.IbisError,
match=r".*Unsupported input type for adjust context.*"):
adjust_context(op, scope, context)
def compute_time_context_window(op,
clients,
timecontext: Optional[TimeContext] = None,
**kwargs):
new_timecontexts = [
timecontext for arg in op.inputs if is_computable_input(arg)
]
if not timecontext:
return new_timecontexts
result = adjust_context(op, timecontext=timecontext)
new_timecontexts = [
result for arg in op.inputs if is_computable_input(arg)
]
return new_timecontexts
def compute_time_context_asof_join(op,
clients,
timecontext: Optional[TimeContext] = None,
**kwargs):
new_timecontexts = [
timecontext for arg in op.inputs if is_computable_input(arg)
]
if not timecontext:
return new_timecontexts
# right table is the second node in children
new_timecontexts = [
new_timecontexts[0],
adjust_context(op, timecontext=timecontext),
*new_timecontexts[2:],
]
return new_timecontexts
def compute_time_context_window(op: ops.WindowOp,
scope: Scope,
clients: List[BaseBackend],
timecontext: Optional[TimeContext] = None,
**kwargs):
new_timecontexts = [
timecontext for arg in op.inputs if is_computable_input(arg)
]
if not timecontext:
return new_timecontexts
result = adjust_context(op, scope, timecontext)
new_timecontexts = [
result for arg in op.inputs if is_computable_input(arg)
]
return new_timecontexts
def compute_time_context_asof_join(op: ops.AsOfJoin,
scope: Scope,
clients: List[BaseBackend],
timecontext: Optional[TimeContext] = None,
**kwargs):
new_timecontexts = [
timecontext for arg in op.inputs if is_computable_input(arg)
]
if not timecontext:
return new_timecontexts
# right table is the second node in children
new_timecontexts = [
new_timecontexts[0],
adjust_context(op, scope, timecontext),
*new_timecontexts[2:],
]
return new_timecontexts