Add tests for padding code

lsst-sitcom · Nov 27, 2023 · d8155ce · d8155ce
1 parent 35a3df9
commit d8155ce
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_efdUtils.py b/tests/test_efdUtils.py
@@ -237,6 +237,18 @@ def test_clipDataToEvent(self):
 
         self.assertGreaterEqual(dataStart, self.event.begin)
         self.assertLessEqual(dataEnd, self.event.end)
+
+        # test the pre/post padding options
+        clippedPaddedData = clipDataToEvent(dayObsData, self.event, prePadding=1, postPadding=2)
+        self.assertIsInstance(clippedPaddedData, pd.DataFrame)
+        self.assertGreater(len(clippedPaddedData), 0)
+        self.assertLess(len(clippedPaddedData), len(dayObsData))
+        self.assertGreater(len(clippedPaddedData), len(clippedData))
+
+        paddedDataStart = efdTimestampToAstropy(clippedPaddedData.iloc[0]['private_efdStamp'])
+        paddedDataEnd = efdTimestampToAstropy(clippedPaddedData.iloc[-1]['private_efdStamp'])
+        self.assertLessEqual(paddedDataStart, dataStart)
+        self.assertGreaterEqual(paddedDataEnd, dataEnd)
         return
 
     def test_getDayObsForTime(self):