Adjust unit test to test wrap of number 9 in customDID

dmwm · Mar 26, 2024 · 6c8e7fb · 6c8e7fb
1 parent 92d7c6e
commit 6c8e7fb
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/python/WMCore_t/MicroService_t/MSPileup_t/MSPileupData_t.py b/test/python/WMCore_t/MicroService_t/MSPileup_t/MSPileupData_t.py
@@ -87,6 +87,18 @@ def testCustomDID(self):
         expect = "/abc/xyz/MINIAOD-V124"
         self.assertTrue(did == expect)
 
+        # test last digit wrap
+        pname = "/abc/xyz/MINIAOD-V19"
+        did = customDID(pname)
+        expect = "/abc/xyz/MINIAOD-V20"
+        self.assertTrue(did == expect)
+
+        # test last digit wrap
+        pname = "/abc/xyz/MINIAOD-V99"
+        did = customDID(pname)
+        expect = "/abc/xyz/MINIAOD-V100"
+        self.assertTrue(did == expect)
+
     def testUpdateTransitionRecord(self):
         """Test the addTransitionRecord function"""
         self.logger.info("test addTransitionRecord function")