wavekit 0.5.4__tar.gz → 0.5.5__tar.gz

{wavekit-0.5.4 → wavekit-0.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavekit
-Version: 0.5.4
+Version: 0.5.5
 Summary: a fundamental package for digital circuit waveform analysis
 License-File: LICENSE
 Author: cxzzzz

{wavekit-0.5.4 → wavekit-0.5.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wavekit"
-version = "0.5.4"
+version = "0.5.5"
 description = "a fundamental package for digital circuit waveform analysis"
 authors = ["cxzzzz <cxz19961010@outlook.com>"]
 readme = "README.md"

{wavekit-0.5.4 → wavekit-0.5.5}/src/wavekit/waveform.py

@@ -130,17 +130,30 @@ class Waveform:
         dense sampled waveform to just the value-change events.
 
         Returns a new :class:`Waveform` where no two adjacent entries have the
-        same value.  :meth:`compress` is an alias for this method.
+        same value.  Behavior matches :func:`numpy.unique_consecutive`.
         """
         if len(self.value) <= 1:
             return self.copy()
-        diff_mask = np.diff(self.value) != 0
-        padded_diff_mask = np.concatenate(([1], diff_mask[:-1], [1]))
-        diff_indices = np.where(padded_diff_mask)[0]
-        return self.take(diff_indices)
+        # Keep first element, then keep elements where value changed from previous
+        mask = np.concatenate(([True], np.diff(self.value) != 0))
+        return self.take(np.where(mask)[0])
 
     def compress(self) -> Waveform:
-        return self.unique_consecutive()
+        """Remove consecutive duplicate values, keeping the last occurrence.
+
+        Unlike :meth:`unique_consecutive`, this preserves the final timestamp
+        and value, which is useful for waveforms where the end time matters.
+
+        Returns a new :class:`Waveform` where no two adjacent entries have the
+        same value, except the last sample is always preserved.
+        """
+        if len(self.value) <= 1:
+            return self.copy()
+        diff_mask = np.diff(self.value) != 0
+        # Keep where next value changed (which means current is last of its group)
+        # plus always keep last element
+        mask = np.concatenate((diff_mask, [True]))
+        return self.take(np.where(mask)[0])
 
     def vectorized_filter(
         self,