Remove if __name__ == "__main__": from test files

smokestacklightnin · smokestacklightnin · commit ab5f4f88f462 · 2024-08-15T20:57:41.000-07:00
diff --git a/tensorflow_transform/analyzers_test.py b/tensorflow_transform/analyzers_test.py
@@ -624,5 +624,3 @@ def testMinDiffFromAvg(self):
         analyzers.calculate_recommended_min_diff_from_avg(100000000), 25)
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/annotators_test.py b/tensorflow_transform/annotators_test.py
@@ -65,5 +65,3 @@ def preprocessing_fn():
     self.assertEqual(trackable_object, object_tracker.trackable_objects[0])
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/beam/analysis_graph_builder_test.py b/tensorflow_transform/beam/analysis_graph_builder_test.py
@@ -579,5 +579,3 @@ class _Analyzer(
                                        structured_outputs)
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/analyzer_cache_test.py b/tensorflow_transform/beam/analyzer_cache_test.py
@@ -310,5 +310,3 @@ def expand(self, pbegin):
             beam_test_util.equal_to([test_cache_dict[key].cache_dict['b']]))
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/beam/analyzer_impls_test.py b/tensorflow_transform/beam/analyzer_impls_test.py
@@ -136,5 +136,3 @@ def testJoinBoundarieRows(self, input_boundaries, expected_boundaries,
     self.assertAllEqual(num_buckets, expected_num_buckets)
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/annotators_test.py b/tensorflow_transform/beam/annotators_test.py
@@ -259,5 +259,3 @@ def preprocessing_fn(inputs):
     )
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/bucketize_integration_test.py b/tensorflow_transform/beam/bucketize_integration_test.py
@@ -888,5 +888,3 @@ def testBucketizationSpecificDistribution(self):
         inputs, expected_boundaries, tf.float32, num_buckets=5)
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/cached_impl_test.py b/tensorflow_transform/beam/cached_impl_test.py
@@ -1749,5 +1749,3 @@ def preprocessing_fn(inputs):
     )
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/combiner_packing_util_test.py b/tensorflow_transform/beam/combiner_packing_util_test.py
@@ -433,5 +433,3 @@ def _side_effect_fn(saved_model_future, cache_value_nodes,
         second=expected_dot_graph_str_after_packing)
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/beam/context_test.py b/tensorflow_transform/beam/context_test.py
@@ -40,5 +40,3 @@ def testNestedContextCreateBaseTempDir(self):
       tft_beam.Context.create_base_temp_dir()
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/deep_copy_test.py b/tensorflow_transform/beam/deep_copy_test.py
@@ -330,5 +330,3 @@ def testDeepCopyTags(self):
     self.assertEqual(DeepCopyTest._counts['Add2'], 3 * (num_copies + 1))
     self.assertEqual(DeepCopyTest._counts['Add3'], 3)
 
-if __name__ == '__main__':
-  unittest.main()
diff --git a/tensorflow_transform/beam/impl_output_record_batches_test.py b/tensorflow_transform/beam/impl_output_record_batches_test.py
@@ -199,5 +199,3 @@ def testConvertToLargeRecordBatch(
     self.assertGreater(actual_num_batches, 1)
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/impl_test.py b/tensorflow_transform/beam/impl_test.py
@@ -4797,5 +4797,3 @@ def test_preprocessing_fn_returns_wrong_type(self):
           expected_data=None)
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/tft_beam_io/beam_metadata_io_test.py b/tensorflow_transform/beam/tft_beam_io/beam_metadata_io_test.py
@@ -92,5 +92,3 @@ def mock_write_metadata(metadata, path):
       self.assertEqual(metadata, test_metadata.COMPLETE_METADATA)
 
 
-if __name__ == '__main__':
-  tf.test.main()
diff --git a/tensorflow_transform/beam/tft_beam_io/transform_fn_io_test.py b/tensorflow_transform/beam/tft_beam_io/transform_fn_io_test.py
@@ -149,5 +149,3 @@ def mock_copy_tree_to_unique_temp_dir(source, base_temp_dir_path):
     self.assertEqual(2, len(file_io.list_directory(transform_output_dir)))
 
 
-if __name__ == '__main__':
-  tf.test.main()
diff --git a/tensorflow_transform/beam/tukey_hh_params_integration_test.py b/tensorflow_transform/beam/tukey_hh_params_integration_test.py
@@ -627,5 +627,3 @@ def assert_and_cast_dtype(tensor):
         # Runs the test deterministically on the whole batch.
         beam_pipeline=beam.Pipeline())
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/vocabulary_integration_test.py b/tensorflow_transform/beam/vocabulary_integration_test.py
@@ -2088,5 +2088,3 @@ def preprocessing_fn(inputs):
     )
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/beam/vocabulary_tfrecord_gzip_integration_test.py b/tensorflow_transform/beam/vocabulary_tfrecord_gzip_integration_test.py
@@ -25,5 +25,3 @@ def _VocabFormat(self):
     return 'tfrecord_gzip'
 
 
-if __name__ == '__main__':
-  tft_unit.main()
diff --git a/tensorflow_transform/coders/csv_coder_test.py b/tensorflow_transform/coders/csv_coder_test.py
@@ -292,5 +292,3 @@ def test_picklable(self):
       self.assertEqual(coder.encode(instance), csv_line.encode('utf-8'))
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/coders/example_proto_coder_test.py b/tensorflow_transform/coders/example_proto_coder_test.py
@@ -406,5 +406,3 @@ def test_example_proto_coder_cache(self):
       self.assertSerializedProtosEqual(coder.encode(instance), serialized_proto)
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/common_test.py b/tensorflow_transform/common_test.py
@@ -69,5 +69,3 @@ def fn2():
     self.assertAllEqual([], graph.get_collection("another_collection"))
 
 
-if __name__ == "__main__":
-  test_case.main()
diff --git a/tensorflow_transform/gaussianization_test.py b/tensorflow_transform/gaussianization_test.py
@@ -252,5 +252,3 @@ def test_inverse_tukey_hh(self, samples, hl, hr, expected_output):
     self.assertAllClose(output, expected_output)
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/graph_tools_test.py b/tensorflow_transform/graph_tools_test.py
@@ -1264,5 +1264,3 @@ def _value_to_matcher(value, add_quotes=False):
       type(value), value))
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/impl_helper_test.py b/tensorflow_transform/impl_helper_test.py
@@ -974,5 +974,3 @@ def iteration(counter, x_minus_counter):
       cond=stop_condition, body=iteration, loop_vars=initial_values)[1]
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/info_theory_test.py b/tensorflow_transform/info_theory_test.py
@@ -166,5 +166,3 @@ def test_mutual_information(self, cell_count, row_count, col_count,
     self.assertNear(per_cell_mi, expected_mi, EPSILON)
 
 
-if __name__ == '__main__':
-  unittest.main()
diff --git a/tensorflow_transform/inspect_preprocessing_fn_test.py b/tensorflow_transform/inspect_preprocessing_fn_test.py
@@ -166,5 +166,3 @@ def test_column_inference(self, preprocessing_fn,
                           expected_transform_input_columns)
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/mappers_test.py b/tensorflow_transform/mappers_test.py
@@ -931,5 +931,3 @@ def testEstimatedProbabilityDensityMissingKey(self):
         self.assertAllEqual(expected, sess.run(result))
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/nodes_test.py b/tensorflow_transform/nodes_test.py
@@ -219,5 +219,3 @@ def testGetDotGraph(self):
         msg='Result dot graph is:\n{}'.format(dot_string))
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/pretrained_models_test.py b/tensorflow_transform/pretrained_models_test.py
@@ -154,5 +154,3 @@ def testApplyFunctionWithCheckpointTwoInputs(self):
         self.assertAllEqual(output_value, [-1, 2, 5])
 
 
-if __name__ == '__main__':
-  tf.test.main()
diff --git a/tensorflow_transform/saved/saved_model_loader_test.py b/tensorflow_transform/saved/saved_model_loader_test.py
@@ -46,5 +46,3 @@ def setUpClass(cls):
 
   # This class has no tests at the moment.
 
-if __name__ == '__main__':
-  unittest.main()
diff --git a/tensorflow_transform/saved/saved_transform_io_test.py b/tensorflow_transform/saved/saved_transform_io_test.py
@@ -305,5 +305,3 @@ def test_stale_asset_collections_are_cleaned(self):
           saved_transform_io.write_saved_transform_from_session(
               session, inputs, outputs, export_path)
 
-if __name__ == '__main__':
-  tf.test.main()
diff --git a/tensorflow_transform/saved/saved_transform_io_v2_test.py b/tensorflow_transform/saved/saved_transform_io_v2_test.py
@@ -666,5 +666,3 @@ def func(inputs):
     self.assertEqual(restored_function(**input_kwargs)['a+b'], expected_output)
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/schema_inference_test.py b/tensorflow_transform/schema_inference_test.py
@@ -402,5 +402,3 @@ def preprocessing_fn(_):
     self.assertProtoEquals(expected_schema, schema)
 
 
-if __name__ == '__main__':
-  unittest.main()
diff --git a/tensorflow_transform/test_case_test.py b/tensorflow_transform/test_case_test.py
@@ -80,5 +80,3 @@ def testSampleParametrizedTestMethod(self, my_arg, my_other_arg):
     self.assertIn((my_arg, my_other_arg), {(1, 'a'), (2, 'b')})
 
 
-if __name__ == '__main__':
-  unittest.main()
diff --git a/tensorflow_transform/tf2_utils_test.py b/tensorflow_transform/tf2_utils_test.py
@@ -111,5 +111,3 @@ def foo(inputs):
     self.assertEqual(result['x_2'].dtype, dtype)
 
 
-if __name__ == '__main__':
-  test_case.main()
diff --git a/tensorflow_transform/tf_metadata/dataset_metadata_test.py b/tensorflow_transform/tf_metadata/dataset_metadata_test.py
@@ -26,5 +26,3 @@ def test_sanity(self):
     self.assertEqual(metadata.schema, test_common.get_test_schema())
 
 
-if __name__ == '__main__':
-  unittest.main()
diff --git a/tensorflow_transform/tf_metadata/metadata_io_test.py b/tensorflow_transform/tf_metadata/metadata_io_test.py
@@ -134,5 +134,3 @@ def test_write_and_read(self):
     self.assertEqual(original, reloaded)
 
 
-if __name__ == '__main__':
-  unittest.main()
diff --git a/tensorflow_transform/tf_metadata/schema_utils_test.py b/tensorflow_transform/tf_metadata/schema_utils_test.py
@@ -80,5 +80,3 @@ def test_pop_ragged_source_columns(self, name, tensor_representation,
     self.assertEqual(feature_by_name, truncated_feature_by_name)
 
 
-if __name__ == '__main__':
-  unittest.main()
diff --git a/tensorflow_transform/tf_utils_test.py b/tensorflow_transform/tf_utils_test.py
@@ -2521,5 +2521,3 @@ def foo(input_tensor):
     self.assertAllEqual(output_tensor, expected_data)
 
 
-if __name__ == '__main__':
-  test_case.main()

Original file line number	Diff line number	Diff line change
`@@ -624,5 +624,3 @@ def testMinDiffFromAvg(self):`
`624`	`624`	`analyzers.calculate_recommended_min_diff_from_avg(100000000), 25)`
`625`	`625`
`626`	`626`
`627`		`-if __name__ == '__main__':`
`628`		`- test_case.main()`
Original file line number	Diff line number	Diff line change
`@@ -65,5 +65,3 @@ def preprocessing_fn():`
`65`	`65`	`self.assertEqual(trackable_object, object_tracker.trackable_objects[0])`
`66`	`66`
`67`	`67`
`68`		`-if __name__ == '__main__':`
`69`		`- test_case.main()`
Original file line number	Diff line number	Diff line change
`@@ -579,5 +579,3 @@ class _Analyzer(`
`579`	`579`	`structured_outputs)`
`580`	`580`
`581`	`581`
`582`		`-if __name__ == '__main__':`
`583`		`- tft_unit.main()`
Original file line number	Diff line number	Diff line change
`@@ -310,5 +310,3 @@ def expand(self, pbegin):`
`310`	`310`	`beam_test_util.equal_to([test_cache_dict[key].cache_dict['b']]))`
`311`	`311`
`312`	`312`
`313`		`-if __name__ == '__main__':`
`314`		`- test_case.main()`
Original file line number	Diff line number	Diff line change
`@@ -136,5 +136,3 @@ def testJoinBoundarieRows(self, input_boundaries, expected_boundaries,`
`136`	`136`	`self.assertAllEqual(num_buckets, expected_num_buckets)`
`137`	`137`
`138`	`138`
`139`		`-if __name__ == '__main__':`
`140`		`- tft_unit.main()`
Original file line number	Diff line number	Diff line change
`@@ -259,5 +259,3 @@ def preprocessing_fn(inputs):`
`259`	`259`	`)`
`260`	`260`
`261`	`261`
`262`		`-if __name__ == '__main__':`
`263`		`- tft_unit.main()`
Original file line number	Diff line number	Diff line change
`@@ -888,5 +888,3 @@ def testBucketizationSpecificDistribution(self):`
`888`	`888`	`inputs, expected_boundaries, tf.float32, num_buckets=5)`
`889`	`889`
`890`	`890`
`891`		`-if __name__ == '__main__':`
`892`		`- tft_unit.main()`
Original file line number	Diff line number	Diff line change
`@@ -1749,5 +1749,3 @@ def preprocessing_fn(inputs):`
`1749`	`1749`	`)`
`1750`	`1750`
`1751`	`1751`
`1752`		`-if __name__ == '__main__':`
`1753`		`- tft_unit.main()`
Original file line number	Diff line number	Diff line change
`@@ -433,5 +433,3 @@ def _side_effect_fn(saved_model_future, cache_value_nodes,`
`433`	`433`	`second=expected_dot_graph_str_after_packing)`
`434`	`434`
`435`	`435`
`436`		`-if __name__ == '__main__':`
`437`		`- test_case.main()`
Original file line number	Diff line number	Diff line change
`@@ -40,5 +40,3 @@ def testNestedContextCreateBaseTempDir(self):`
`40`	`40`	`tft_beam.Context.create_base_temp_dir()`
`41`	`41`
`42`	`42`
`43`		`-if __name__ == '__main__':`
`44`		`- tft_unit.main()`