select default col in test

Author: BryanCutler

python/pyspark/ml/tests.py

@@ -664,7 +664,8 @@ def test_count_vectorizer_from_vocab(self):
 
         # Test model with default settings can transform
         model_default = CountVectorizerModel.from_vocabulary(["a", "b", "c"], inputCol="words")
-        transformed_list = model_default.transform(dataset).collect()
+        transformed_list = model_default.transform(dataset)\
+            .select(model_default.getOrDefault(model_default.outputCol)).collect()
         self.assertEqual(len(transformed_list), 3)
 
     def test_rformula_force_index_label(self):