def single_output_expression(self) -> str:
assert_or_throw(
len(self.outputs) == 1,
lambda: FugueWorkflowError(f"{self.name} does not have single output"),
)
return self.name + "." + self.outputs.get_key_by_index(0)
def process(self, dfs: DataFrames) -> DataFrame:
assert_or_throw(len(dfs) == 1, FugueWorkflowError("not single input"))
columns = self.params.get_or_throw("columns", list)
return dfs[0][columns]
def process(self, dfs: DataFrames) -> DataFrame:
assert_or_throw(len(dfs) == 1, FugueWorkflowError("not single input"))
return self.execution_engine.distinct(dfs[0])
def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
raise FugueWorkflowError( # pragma: no cover
"with statement is invalid for _FugueInteractiveWorkflow"
)
def __enter__(self):
raise FugueWorkflowError(
"with statement is invalid for _FugueInteractiveWorkflow"
)
def process(self, dfs: DataFrames) -> None:
assert_or_throw(
len(dfs) > 1, FugueWorkflowError("can't accept single input"))
expected = dfs[0]
for i in range(1, len(dfs)):
assert not _df_eq(expected, dfs[i], throw=False, **self.params)
