예제 #1
0
 def test_transforms_fetching_two_columns(self):
     ce = ColumnExtractor(['trecho', 'palavra'])
     actual = ce.transform(X)
     expected = [['a', 'b'], ['c', 'd']]
     self.assertEquals(expected, actual)
예제 #2
0
 def test_transforms_fetching_one_attribute_column(self):
     ce = ColumnExtractor(['Child.prop'])
     actual = ce.transform(X)
     expected = [['w'], ['z']]
     self.assertEquals(expected, actual)
예제 #3
0
 def test_transforms_fetching_only_one_column(self):
     ce = ColumnExtractor(['trecho'])
     actual = ce.transform(X)
     expected = [['a'], ['c']]
     self.assertEquals(expected, actual)