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)