Skip to content

Commit

Permalink
Added matroushka sample and test case
Browse files Browse the repository at this point in the history
  • Loading branch information
CarterFendley committed Sep 20, 2024
1 parent 217ff0d commit 0b6b09d
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions samples/v2/sample_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
import producer_consumer_param
import pipeline_container_no_input
import two_step_pipeline_containerized
import matroushka

_MINUTE = 60 # seconds
_DEFAULT_TIMEOUT = 5 * _MINUTE
Expand All @@ -50,6 +51,7 @@ def test(self):
TestCase(pipeline_func=two_step_pipeline_containerized.two_step_pipeline_containerized),
TestCase(pipeline_func=component_with_optional_inputs.pipeline),
TestCase(pipeline_func=pipeline_with_env.pipeline_with_env),
TestCase(pipeline_func=matroushka.large_matroushka_doll)

# The following tests are not working. Tracking issue: https://github.com/kubeflow/pipelines/issues/11053
# TestCase(pipeline_func=pipeline_with_importer.pipeline_with_importer),
Expand Down Expand Up @@ -77,6 +79,7 @@ def run_test_case(self, pipeline_func: GraphComponent, timeout: int):
print(f"{self._kfp_ui_and_port}/#/runs/details/{run_response.run_id}")

self.assertEqual(run_response.state, "SUCCEEDED")
print(f"Pipeline succeeded: {pipeline_func.pipeline_func.__name__}")


if __name__ == '__main__':
Expand Down

0 comments on commit 0b6b09d

Please sign in to comment.