Remove if __name__ == "__main__" from test files because it is unused with pytest

Author: smokestacklightnin

tensorflow_model_analysis/api/dataframe_test.py

@@ -516,5 +516,3 @@ def testAutoPivot_PlotsDataFrameCollapseColumnNames(self):
     )
     pd.testing.assert_frame_equal(expected, df, check_column_type=False)
 
-if __name__ == '__main__':
-  tf.test.main()

tensorflow_model_analysis/api/model_eval_lib_test.py

@@ -1579,6 +1579,3 @@ def testBytesProcessedCountForRecordBatches(self):
     self.assertEqual(actual_counter[0].committed, expected_num_bytes)
 
 
-if __name__ == '__main__':
-  tf.compat.v1.enable_v2_behavior()
-  tf.test.main()

tensorflow_model_analysis/api/types_test.py

@@ -91,5 +91,3 @@ def testVarLenTensorValueEmpty(self):
     )
 
 
-if __name__ == '__main__':
-  absltest.main()

tensorflow_model_analysis/contrib/aggregates/binary_confusion_matrices_test.py

@@ -300,5 +300,3 @@ def testBinaryConfusionMatricesInProcess(
     self.assertDictEqual(actual, expected_result)
 
 
-if __name__ == '__main__':
-  tf.test.main()

tensorflow_model_analysis/evaluators/analysis_table_evaluator_test.py

@@ -93,5 +93,3 @@ def check_result(got):
       util.assert_that(got[constants.ANALYSIS_KEY], check_result)
 
 
-if __name__ == '__main__':
-  tf.test.main()

tensorflow_model_analysis/evaluators/confidence_intervals_util_test.py

@@ -325,5 +325,3 @@ def check_result(got_pcoll):
       util.assert_that(result, check_result)
 
 
-if __name__ == '__main__':
-  absltest.main()

tensorflow_model_analysis/evaluators/counter_util_test.py

@@ -69,5 +69,3 @@ def testMetricsSpecBeamCounter(self):
       self.assertEqual(actual_metrics_count, 1)
 
 
-if __name__ == '__main__':
-  tf.test.main()

tensorflow_model_analysis/evaluators/evaluator_test.py

@@ -45,5 +45,3 @@ def testVerifyEvaluatorRaisesValueError(self):
       )
 
 
-if __name__ == '__main__':
-  tf.test.main()

tensorflow_model_analysis/evaluators/jackknife_test.py

@@ -272,5 +272,3 @@ def check_result(got_pcoll):
       util.assert_that(result, check_result)
 
 
-if __name__ == '__main__':
-  absltest.main()

tensorflow_model_analysis/evaluators/legacy_poisson_bootstrap_test.py

@@ -90,5 +90,3 @@ def testCalculateConfidenceInterval(self):
     )
 
 
-if __name__ == '__main__':
-  tf.test.main()