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_smoke_test_table_three():
spark = spark_session_from_config()
result = invoke_compute(
TableThree,
inputs={
'table_one': spark.createDataFrame([Row(num=1)]),
'table_two': spark.createDataFrame([Row(num=2)]),
},
)
assert result.success
assert set(result.output_value().collect()) == set([Row(num=1), Row(num=2)])