Beispiel #1
0
 def test_verify_input_artifacts_not_exists(self):
     with self.assertRaises(RuntimeError):
         base_driver._verify_input_artifacts(
             {'artifact': [types.TfxArtifact(type_name='input_data')]})
Beispiel #2
0
 def testVerifyInputArtifactsNotExists(self):
   with self.assertRaises(RuntimeError):
     base_driver._verify_input_artifacts(
         {'artifact': [types.Artifact(type_name='input_data')]})
Beispiel #3
0
 def test_verify_input_artifacts_ok(self):
     base_driver._verify_input_artifacts(self._input_artifacts)
Beispiel #4
0
 def testVerifyInputArtifactsOk(self):
   base_driver._verify_input_artifacts(self._input_artifacts)