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