Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_execute_in_mem_lakehouse(execute_spark_lakehouse_build):
lakehouse = PySparkMemLakehouse()
pipeline_result = execute_spark_lakehouse_build(
tables=[TableOne, TableTwo, TableThree],
lakehouse=lakehouse,
environment_dict={'solids': {'TableOne': {'inputs': {'num': {'value': 1}}}}},
)
assert pipeline_result.success
assert lakehouse.collected_tables == {
'TableOne': [Row(num=1)],
'TableTwo': [Row(num=2)],
'TableThree': [Row(num=1), Row(num=2)],
}