Add missing v2_behavior fixtures

Author: smokestacklightnin

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
 ):

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
 ):

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
 ):

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
 ):

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
 ):

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
 ):

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
 ):

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
 ):

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,

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):