Skip to content

Commit 6afb844

Browse files
tensorflower-gardenerTF Object Detection Team
authored andcommitted
cleanup of deprecated test methods
PiperOrigin-RevId: 725124395
1 parent f43a1c6 commit 6afb844

12 files changed

+34
-34
lines changed

research/object_detection/anchor_generators/flexible_grid_anchor_generator_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def test_construct_normalized_anchors_fails_with_unit_dimensions(self):
7474
anchor_generator = fg.FlexibleGridAnchorGenerator(
7575
[(32.0,)], [(1.0,)], [(32, 32),], [(16, 16),],
7676
normalize_coordinates=True)
77-
with self.assertRaisesRegexp(ValueError, 'Normalized coordinates'):
77+
with self.assertRaisesRegex(ValueError, 'Normalized coordinates'):
7878
anchor_generator.generate(
7979
feature_map_shape_list=[(2, 2)], im_height=1, im_width=1)
8080

research/object_detection/anchor_generators/multiscale_grid_anchor_generator_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def test_construct_normalized_anchors_fails_with_unit_dimensions(self):
7878
anchor_generator = mg.MultiscaleGridAnchorGenerator(
7979
min_level=5, max_level=5, anchor_scale=1.0, aspect_ratios=[1.0],
8080
scales_per_octave=1, normalize_coordinates=True)
81-
with self.assertRaisesRegexp(ValueError, 'Normalized coordinates'):
81+
with self.assertRaisesRegex(ValueError, 'Normalized coordinates'):
8282
anchor_generator.generate(
8383
feature_map_shape_list=[(2, 2)], im_height=1, im_width=1)
8484

research/object_detection/core/prefetcher_tf1_test.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,9 +60,9 @@ def test_prefetch_tensors_with_fully_defined_shapes(self):
6060
with slim.queues.QueueRunners(sess):
6161
for _ in range(num_batches):
6262
results = sess.run(tensor_dict)
63-
self.assertEquals(results['image'].shape,
63+
self.assertEqual(results['image'].shape,
6464
(batch_size, image_size, image_size, 3))
65-
self.assertEquals(results['label'].shape, (batch_size, 1))
65+
self.assertEqual(results['label'].shape, (batch_size, 1))
6666
with self.assertRaises(tf.errors.OutOfRangeError):
6767
sess.run(tensor_dict)
6868

@@ -98,9 +98,9 @@ def test_prefetch_tensors_with_partially_defined_shapes(self):
9898
with slim.queues.QueueRunners(sess):
9999
for _ in range(num_batches):
100100
results = sess.run(tensor_dict)
101-
self.assertEquals(results['image'].shape,
101+
self.assertEqual(results['image'].shape,
102102
(batch_size, image_size, image_size, 3))
103-
self.assertEquals(results['label'].shape, (batch_size, 1))
103+
self.assertEqual(results['label'].shape, (batch_size, 1))
104104
with self.assertRaises(tf.errors.OutOfRangeError):
105105
sess.run(tensor_dict)
106106

research/object_detection/core/target_assigner_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -462,7 +462,7 @@ def test_raises_error_on_incompatible_groundtruth_boxes_and_labels(self):
462462
groundtruth_labels = tf.constant([[0, 1, 0, 0, 0, 0, 0],
463463
[0, 0, 0, 0, 0, 1, 0],
464464
[0, 0, 0, 1, 0, 0, 0]], tf.float32)
465-
with self.assertRaisesRegexp(ValueError, 'Unequal shapes'):
465+
with self.assertRaisesRegex(ValueError, 'Unequal shapes'):
466466
target_assigner.assign(
467467
priors,
468468
boxes,

research/object_detection/exporter_tf1_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -665,7 +665,7 @@ def test_raise_runtime_error_on_images_with_different_sizes(self):
665665
keypoints = inference_graph.get_tensor_by_name('detection_keypoints:0')
666666
masks = inference_graph.get_tensor_by_name('detection_masks:0')
667667
num_detections = inference_graph.get_tensor_by_name('num_detections:0')
668-
with self.assertRaisesRegexp(tf.errors.InvalidArgumentError,
668+
with self.assertRaisesRegex(tf.errors.InvalidArgumentError,
669669
'TensorArray.*shape'):
670670
sess.run(
671671
[boxes, scores, classes, keypoints, masks, num_detections],

research/object_detection/model_lib_tf1_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ def test_model_fn_in_train_mode_freeze_all_variables(self):
226226
"""Tests model_fn TRAIN mode with all variables frozen."""
227227
configs = _get_configs_for_model(MODEL_NAME_FOR_TEST)
228228
configs['train_config'].freeze_variables.append('.*')
229-
with self.assertRaisesRegexp(ValueError, 'No variables to optimize'):
229+
with self.assertRaisesRegex(ValueError, 'No variables to optimize'):
230230
self._assert_model_fn_for_train_eval(configs, 'train')
231231

232232
def test_model_fn_in_train_mode_freeze_all_included_variables(self):
@@ -235,7 +235,7 @@ def test_model_fn_in_train_mode_freeze_all_included_variables(self):
235235
train_config = configs['train_config']
236236
train_config.update_trainable_variables.append('FeatureExtractor')
237237
train_config.freeze_variables.append('.*')
238-
with self.assertRaisesRegexp(ValueError, 'No variables to optimize'):
238+
with self.assertRaisesRegex(ValueError, 'No variables to optimize'):
239239
self._assert_model_fn_for_train_eval(configs, 'train')
240240

241241
def test_model_fn_in_train_mode_freeze_box_predictor(self):

research/object_detection/utils/config_util_test.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -866,22 +866,22 @@ def testCheckAndParseInputConfigKey(self):
866866
self.assertEqual(input_name, None)
867867
self.assertEqual(field_name, "label_map_path")
868868

869-
with self.assertRaisesRegexp(ValueError,
869+
with self.assertRaisesRegex(ValueError,
870870
"Invalid key format when overriding configs."):
871871
config_util.check_and_parse_input_config_key(
872872
configs, "train_input_config:shuffle")
873873

874-
with self.assertRaisesRegexp(
874+
with self.assertRaisesRegex(
875875
ValueError, "Invalid key_name when overriding input config."):
876876
config_util.check_and_parse_input_config_key(
877877
configs, "invalid_key_name:train_name:shuffle")
878878

879-
with self.assertRaisesRegexp(
879+
with self.assertRaisesRegex(
880880
ValueError, "Invalid input_name when overriding input config."):
881881
config_util.check_and_parse_input_config_key(
882882
configs, "eval_input_configs:unknown_eval_name:shuffle")
883883

884-
with self.assertRaisesRegexp(
884+
with self.assertRaisesRegex(
885885
ValueError, "Invalid field_name when overriding input config."):
886886
config_util.check_and_parse_input_config_key(
887887
configs, "eval_input_configs:eval_2:unknown_field_name")
@@ -919,7 +919,7 @@ def testUpdateInputReaderConfigErrors(self):
919919
_write_config(pipeline_config, pipeline_config_path)
920920
configs = config_util.get_configs_from_pipeline_file(pipeline_config_path)
921921

922-
with self.assertRaisesRegexp(ValueError,
922+
with self.assertRaisesRegex(ValueError,
923923
"Duplicate input name found when overriding."):
924924
config_util.update_input_reader_config(
925925
configs,
@@ -928,7 +928,7 @@ def testUpdateInputReaderConfigErrors(self):
928928
field_name="shuffle",
929929
value=False)
930930

931-
with self.assertRaisesRegexp(
931+
with self.assertRaisesRegex(
932932
ValueError, "Input name name_not_exist not found when overriding."):
933933
config_util.update_input_reader_config(
934934
configs,
@@ -937,7 +937,7 @@ def testUpdateInputReaderConfigErrors(self):
937937
field_name="shuffle",
938938
value=False)
939939

940-
with self.assertRaisesRegexp(ValueError,
940+
with self.assertRaisesRegex(ValueError,
941941
"Unknown input config overriding."):
942942
config_util.update_input_reader_config(
943943
configs,

research/object_detection/utils/metrics_test.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,8 @@ def test_compute_precision_recall_and_ap_no_groundtruth(self):
9191
expected_precision = None
9292
expected_recall = None
9393
precision, recall = metrics.compute_precision_recall(scores, labels, num_gt)
94-
self.assertEquals(precision, expected_precision)
95-
self.assertEquals(recall, expected_recall)
94+
self.assertEqual(precision, expected_precision)
95+
self.assertEqual(recall, expected_recall)
9696
ap = metrics.compute_average_precision(precision, recall)
9797
self.assertTrue(np.isnan(ap))
9898

@@ -137,10 +137,10 @@ def test_compute_median_rank_at_k(self):
137137
median_ranks_5 = metrics.compute_median_rank_at_k(tp_fp, 5)
138138
median_ranks_3_bool = metrics.compute_median_rank_at_k(tp_fp_bool, 3)
139139

140-
self.assertEquals(median_ranks_1, 0)
141-
self.assertEquals(median_ranks_3, 0.5)
142-
self.assertEquals(median_ranks_3_bool, 0.5)
143-
self.assertEquals(median_ranks_5, 0.5)
140+
self.assertEqual(median_ranks_1, 0)
141+
self.assertEqual(median_ranks_3, 0.5)
142+
self.assertEqual(median_ranks_3_bool, 0.5)
143+
self.assertEqual(median_ranks_5, 0.5)
144144

145145

146146
if __name__ == '__main__':

research/object_detection/utils/np_box_list_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def test_num_boxes(self):
132132
boxes = np.array([[0., 0., 100., 100.], [10., 30., 50., 70.]], dtype=float)
133133
boxlist = np_box_list.BoxList(boxes)
134134
expected_num_boxes = 2
135-
self.assertEquals(boxlist.num_boxes(), expected_num_boxes)
135+
self.assertEqual(boxlist.num_boxes(), expected_num_boxes)
136136

137137

138138
if __name__ == '__main__':

research/object_detection/utils/np_box_mask_list_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def test_num_boxes(self):
179179
box_mask_list = np_box_mask_list.BoxMaskList(
180180
box_data=boxes, mask_data=masks)
181181
expected_num_boxes = 2
182-
self.assertEquals(box_mask_list.num_boxes(), expected_num_boxes)
182+
self.assertEqual(box_mask_list.num_boxes(), expected_num_boxes)
183183

184184

185185
if __name__ == '__main__':

0 commit comments

Comments
 (0)