Example #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')]})
Example #2
0
 def testVerifyInputArtifactsNotExists(self):
   with self.assertRaises(RuntimeError):
     base_driver._verify_input_artifacts(
         {'artifact': [types.Artifact(type_name='input_data')]})
Example #3
0
 def test_verify_input_artifacts_ok(self):
     base_driver._verify_input_artifacts(self._input_artifacts)
Example #4
0
 def testVerifyInputArtifactsOk(self):
   base_driver._verify_input_artifacts(self._input_artifacts)