add data_preprocessor ut

open-mmlab · Nov 17, 2022 · 8fed7f5 · 8fed7f5
1 parent d33af52
commit 8fed7f5
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/test_models/test_data_preprocessor.py b/tests/test_models/test_data_preprocessor.py
@@ -46,3 +46,19 @@ def test_forward(self):
  out = processor(data, training=True)
  self.assertEqual(out['inputs'].shape, (2, 3, 20, 20))
  self.assertEqual(len(out['data_samples']), 2)
+
+ # test predict with padding
+ processor = SegDataPreProcessor(
+ mean=[0, 0, 0],
+ std=[1, 1, 1],
+ size=(20, 20),
+ test_cfg=dict(size_divisor=15))
+ data = {
+ 'inputs': [
+ torch.randint(0, 256, (3, 11, 10)),
+ ],
+ 'data_samples': [data_sample]
+ }
+ out = processor(data, training=False)
+ self.assertEqual(out['inputs'].shape[2] % 15, 0)
+ self.assertEqual(out['inputs'].shape[3] % 15, 0)