Add missing v2_behavior fixtures

Author: smokestacklightnin

Diff for: tensorflow_model_analysis/api/model_eval_lib_test.py

@@ -70,6 +70,7 @@
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class EvaluateTest(
     test_util.TensorflowModelAnalysisTest, parameterized.TestCase
 ):

Diff for: tensorflow_model_analysis/evaluators/metrics_validator_test.py

@@ -135,7 +135,7 @@
     ((('feature1', 'value1'),), (('feature3', 'value3'),)),
 )
 
-
+@pytest.mark.usefixtures("v2_behavior")
 class MetricsValidatorTest(
     test_util.TensorflowModelAnalysisTest, parameterized.TestCase
 ):

Diff for: tensorflow_model_analysis/extractors/counterfactual_predictions_extractor_test.py

@@ -55,6 +55,7 @@ def call(self, serialized_example):
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class CounterfactualPredictionsExtactorTest(
     test_util.TensorflowModelAnalysisTest, parameterized.TestCase
 ):

Diff for: tensorflow_model_analysis/extractors/materialized_predictions_extractor_test.py

@@ -35,6 +35,7 @@
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class MaterializedPredictionsExtractorTest(
     testutil.TensorflowModelAnalysisTest
 ):

Diff for: tensorflow_model_analysis/extractors/predictions_extractor_test.py

@@ -39,6 +39,7 @@
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class PredictionsExtractorTest(
     test_util.TensorflowModelAnalysisTest, parameterized.TestCase
 ):

Diff for: tensorflow_model_analysis/extractors/tfjs_predict_extractor_test.py

@@ -44,6 +44,7 @@
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class TFJSPredictExtractorTest(
     testutil.TensorflowModelAnalysisTest, parameterized.TestCase
 ):

Diff for: tensorflow_model_analysis/extractors/tflite_predict_extractor_test.py

@@ -54,6 +54,7 @@ def random_genenerator():
     yield r
 
 
+@pytest.mark.usefixtures("v2_behavior")
 class TFLitePredictExtractorTest(
     testutil.TensorflowModelAnalysisTest, parameterized.TestCase
 ):

Diff for: tensorflow_model_analysis/extractors/transformed_features_extractor_test.py

@@ -40,6 +40,7 @@
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class TransformedFeaturesExtractorTest(
     testutil.TensorflowModelAnalysisTest, parameterized.TestCase
 ):

Diff for: tensorflow_model_analysis/metrics/confusion_matrix_metrics_test.py

@@ -38,6 +38,7 @@
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class ConfusionMatrixMetricsTest(
     test_util.TensorflowModelAnalysisTest,
     metric_test_util.TestCase,

Diff for: tensorflow_model_analysis/metrics/tf_metric_accumulators_test.py

@@ -20,7 +20,7 @@
 from tensorflow_model_analysis.utils import test_util
 
 
-
+@pytest.mark.usefixtures("v2_behavior")
 class TfMetricAccumulatorsTest(test_util.TensorflowModelAnalysisTest):
 
   def testTFMetricsAccumulator(self):

Diff for: tensorflow_model_analysis/metrics/tf_metric_wrapper_test.py

@@ -72,6 +72,7 @@ def result(self):
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class ConfusionMatrixMetricsTest(
     test_util.TensorflowModelAnalysisTest, parameterized.TestCase
 ):
@@ -495,6 +496,7 @@ def check_result(got):
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class NonConfusionMatrixMetricsTest(
     test_util.TensorflowModelAnalysisTest, parameterized.TestCase
 ):
@@ -1047,6 +1049,7 @@ def testMergeAccumulators(self):
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class MixedMetricsTest(test_util.TensorflowModelAnalysisTest):
 
   def testWithMixedMetrics(self):

Diff for: tensorflow_model_analysis/utils/model_util_test.py

@@ -50,6 +50,7 @@ def _record_batch_to_extracts(record_batch):
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class ModelUtilTest(
     test_util.TensorflowModelAnalysisTest, parameterized.TestCase
 ):

Diff for: tensorflow_model_analysis/utils/util_test.py

@@ -26,7 +26,7 @@
 from tensorflow_metadata.proto.v0 import schema_pb2
 
 
-
+@pytest.mark.usefixtures("v2_behavior")
 class UtilTest(tf.test.TestCase, parameterized.TestCase):
 
   def testToTensorValueFromTFSparseTensor(self):

Diff for: tensorflow_model_analysis/writers/metrics_plots_and_validations_writer_test.py

@@ -64,6 +64,7 @@ def _make_slice_key(*args):
 
 @pytest.mark.xfail(run=False, reason="PR 183 This class contains tests that fail and needs to be fixed. "
 "If all tests pass, please remove this mark.")
+@pytest.mark.usefixtures("v2_behavior")
 class MetricsPlotsAndValidationsWriterTest(testutil.TensorflowModelAnalysisTest,
                                            parameterized.TestCase):