Add xfail mark to classes with failing tests

tensorflow_transform/beam/bucketize_integration_test.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 """Tests for tft.bucketize and tft.quantiles."""
 
+
+import pytest
 import contextlib
 import random
 
@@ -339,6 +341,8 @@ def _compute_simple_per_key_bucket(val, key, weighted=False):
 ]
 
 
+@pytest.mark.xfail(reason="PR 315 This class contains tests that fail and needs to be fixed. "
+"If all tests pass, please remove this mark.")
 class BucketizeIntegrationTest(tft_unit.TransformTestCase):
 
   def setUp(self):

tensorflow_transform/beam/impl_output_record_batches_test.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 """Same as impl_test.py, except that impl produces `pa.RecordBatch`es."""
 
+
+import pytest
 import collections
 
 import numpy as np
@@ -28,6 +30,8 @@
 _LARGE_BATCH_SIZE = 1 << 10
 
 
+@pytest.mark.xfail(reason="PR 315 This class contains tests that fail and needs to be fixed. "
+"If all tests pass, please remove this mark.")
 class BeamImplOutputRecordBatchesTest(impl_test.BeamImplTest):
 
   def _OutputRecordBatches(self):

tensorflow_transform/beam/impl_test.py

@@ -14,6 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+
+import pytest
 import itertools
 import math
 import os
@@ -110,6 +112,8 @@ def _mean_output_dtype(input_dtype):
   return tf.float64 if input_dtype == tf.float64 else tf.float32
 
 
+@pytest.mark.xfail(reason="PR 315 This class contains tests that fail and needs to be fixed. "
+"If all tests pass, please remove this mark.")
 class BeamImplTest(tft_unit.TransformTestCase):
 
   def setUp(self):

tensorflow_transform/beam/tukey_hh_params_integration_test.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 """Tests for tft.tukey_* calls (Tukey HH parameters)."""
 
+
+import pytest
 import itertools
 
 import apache_beam as beam
@@ -92,6 +94,8 @@
 ]
 
 
+@pytest.mark.xfail(reason="PR 315 This class contains tests that fail and needs to be fixed. "
+"If all tests pass, please remove this mark.")
 class TukeyHHParamsIntegrationTest(tft_unit.TransformTestCase):
 
   def setUp(self):

tensorflow_transform/beam/vocabulary_integration_test.py

@@ -15,6 +15,8 @@
 # limitations under the License.
 """Tests for tft.vocabulary and tft.compute_and_apply_vocabulary."""
 
+
+import pytest
 import os
 
 import apache_beam as beam
@@ -114,6 +116,8 @@
 ]
 
 
+@pytest.mark.xfail(reason="PR 315 This class contains tests that fail and needs to be fixed. "
+"If all tests pass, please remove this mark.")
 class VocabularyIntegrationTest(tft_unit.TransformTestCase):
 
   def setUp(self):

tensorflow_transform/beam/vocabulary_tfrecord_gzip_integration_test.py

@@ -14,10 +14,14 @@
 # limitations under the License.
 """Tests for tfrecord_gzip tft.vocabulary and tft.compute_and_apply_vocabulary."""
 
+
+import pytest
 from tensorflow_transform.beam import vocabulary_integration_test
 from tensorflow_transform.beam import tft_unit
 
 
+@pytest.mark.xfail(reason="PR 315 This class contains tests that fail and needs to be fixed. "
+"If all tests pass, please remove this mark.")
 class TFRecordVocabularyIntegrationTest(
     vocabulary_integration_test.VocabularyIntegrationTest):
 

tensorflow_transform/tf_utils_test.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 """Tests for tensorflow_transform.tf_utils."""
 
+
+import pytest
 import os
 
 import numpy as np
@@ -65,6 +67,8 @@ def __init__(self, shape, dtype):
   tf.SparseTensorSpec = _SparseTensorSpec
 
 
+@pytest.mark.xfail(reason="PR 315 This class contains tests that fail and needs to be fixed. "
+"If all tests pass, please remove this mark.")
 class TFUtilsTest(test_case.TransformTestCase):
 
   def _assertCompositeRefEqual(self, left, right):