Python load_pipeline Examples

Example #1

File: pipeline_test.py Project: rachthree/magenta_cs230_win19

    def testPipelineIterator(self):
        strings = ['abcdefg', 'helloworld!', 'qwerty']
        result = pipeline.load_pipeline(MockPipeline(), iter(strings))

        self.assertEqual(
            set([MockStringProto(s + '_A') for s in strings] +
                [MockStringProto(s + '_B') for s in strings]),
            set(result['dataset_1']))
        self.assertEqual(set(MockStringProto(s + '_C') for s in strings),
                         set(result['dataset_2']))

Example #2

File: pipeline_test.py Project: Alice-ren/magenta

  def testPipelineIterator(self):
    strings = ['abcdefg', 'helloworld!', 'qwerty']
    result = pipeline.load_pipeline(MockPipeline(), iter(strings))

    self.assertEqual(
        set([MockStringProto(s + '_A') for s in strings] +
            [MockStringProto(s + '_B') for s in strings]),
        set(result['dataset_1']))
    self.assertEqual(
        set([MockStringProto(s + '_C') for s in strings]),
        set(result['dataset_2']))