Esempio n. 1
0
    def testSimpleEnd2EndPipeline(self):
        """End-to-End test for simple pipeline."""
        pipeline_name = 'kubeflow-e2e-test-{}'.format(self._random_id())
        components = test_utils.create_e2e_components(
            self._pipeline_root(pipeline_name), self._data_root,
            self._taxi_module_file)
        pipeline = self._create_pipeline(pipeline_name, components)

        self._compile_and_run_pipeline(pipeline)
Esempio n. 2
0
    def testSimpleEnd2EndPipeline(self):
        """End-to-End test for simple pipeline."""
        pipeline_name = 'kubeflow-e2e-test-{}'.format(test_utils.random_id())
        components = kubeflow_test_utils.create_e2e_components(
            self._pipeline_root(pipeline_name),
            self._data_root,
            self._transform_module,
            self._trainer_module,
        )
        pipeline = self._create_pipeline(pipeline_name, components)

        self._compile_and_run_pipeline(pipeline)
        self._assert_infra_validator_passed(pipeline_name)