add data_preprocessor ut

Author: xiexinch

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)