def testEnableCache(self):
input_base = standard_artifacts.ExternalArtifact()
import_example_gen_1 = component.ImportExampleGen(
input=channel_utils.as_channel([input_base]))
self.assertEqual(None, import_example_gen_1.enable_cache)
import_example_gen_2 = component.ImportExampleGen(
input=channel_utils.as_channel([input_base]), enable_cache=True)
self.assertEqual(True, import_example_gen_2.enable_cache)
def test_construct(self):
input_base = types.Artifact(type_name='ExternalPath')
import_example_gen = component.ImportExampleGen(
input_base=channel.as_channel([input_base]))
self.assertEqual('ExamplesPath',
import_example_gen.outputs.examples.type_name)
artifact_collection = import_example_gen.outputs.examples.get()
self.assertEqual('train', artifact_collection[0].split)
self.assertEqual('eval', artifact_collection[1].split)
def testConstruct(self):
import_example_gen = component.ImportExampleGen(input_base='path')
self.assertEqual(standard_artifacts.Examples.TYPE_NAME,
import_example_gen.outputs['examples'].type_name)
artifact_collection = import_example_gen.outputs['examples'].get()
self.assertEqual(1, len(artifact_collection))
self.assertEqual(['train', 'eval'],
artifact_utils.decode_split_names(
artifact_collection[0].split_names))
def testConstruct(self):
input_base = standard_artifacts.ExternalArtifact()
import_example_gen = component.ImportExampleGen(
input_base=channel_utils.as_channel([input_base]))
self.assertEqual('ExamplesPath',
import_example_gen.outputs['examples'].type_name)
artifact_collection = import_example_gen.outputs['examples'].get()
self.assertEqual('train', artifact_collection[0].split)
self.assertEqual('eval', artifact_collection[1].split)
def testConstruct(self):
import_example_gen = component.ImportExampleGen(input_base='path')
self.assertEqual(standard_artifacts.Examples.TYPE_NAME,
import_example_gen.outputs['examples'].type_name)
