pyrauli 0.3.0__tar.gz → 0.3.1__tar.gz

{pyrauli-0.3.0 → pyrauli-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pyrauli
-Version: 0.3.0
+Version: 0.3.1
 Summary: A very fast and easy to use Quantum circuit simulator relying on Pauli propagation. Compatible with qiskit.
 License:                     GNU GENERAL PUBLIC LICENSE
                                 Version 3, 29 June 2007

{pyrauli-0.3.0 → pyrauli-0.3.1}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "scikit_build_core.build"
 
 [project]
 name = "pyrauli"
-version = "0.3.0"
+version = "0.3.1"
 description = "A very fast and easy to use Quantum circuit simulator relying on Pauli propagation. Compatible with qiskit."
 readme = "README.md"
 requires-python = ">=3.9"

{pyrauli-0.3.0 → pyrauli-0.3.1}/src/pyrauli/__init__.py

@@ -8,7 +8,7 @@ simulation, powered by the C++ `propauli` library.
 from ._core import (
     PauliEnum, PauliGate, CliffordGate, UnitalNoise, QGate, Pauli, PauliTerm,
     Observable, Noise, NoiseModel, Truncator, CoefficientTruncator,
-    WeightTruncator, NeverTruncator, LambdaTruncator, MultiTruncator,
+    WeightTruncator, KeepNTruncator, NeverTruncator, LambdaTruncator, MultiTruncator,
     SchedulingPolicy, NeverPolicy, AlwaysBeforeSplittingPolicy,
     AlwaysAfterSplittingPolicy, Circuit, OperationType, Timing,
     SimulationState, CompressionResult, LambdaPolicy,
@@ -17,7 +17,7 @@ from ._core import (
 __all__ = [
     "PauliEnum", "PauliGate", "CliffordGate", "UnitalNoise", "QGate", "Pauli",
     "PauliTerm", "Observable", "Noise", "NoiseModel", "Truncator",
-    "CoefficientTruncator", "WeightTruncator", "NeverTruncator",
+    "CoefficientTruncator", "WeightTruncator", "KeepNTruncator", "NeverTruncator",
     "LambdaTruncator", "MultiTruncator", "SchedulingPolicy", "NeverPolicy",
     "AlwaysBeforeSplittingPolicy", "AlwaysAfterSplittingPolicy", "Circuit",
     "OperationType", "Timing", "SimulationState", "CompressionResult",

{pyrauli-0.3.0 → pyrauli-0.3.1}/tests/test_truncator.py

@@ -6,6 +6,7 @@ from pyrauli import (
     Circuit,
     CoefficientTruncator,
     WeightTruncator,
+    KeepNTruncator,
     LambdaTruncator,
     MultiTruncator,
     AlwaysAfterSplittingPolicy
@@ -37,6 +38,19 @@ def test_weight_truncator_on_observable():
     assert len(obs) == 1
     assert obs[0] == PauliTerm("IIII", 0.5)
 
+def test_keepn_truncator_on_observable():
+    """Tests that terms with high Pauli weight are removed."""
+    obs = Observable([PauliTerm("IIII", 0.5), PauliTerm("ZYXI", 0.11), PauliTerm("YYYY", -0.5), PauliTerm("IXYZ", 0.1)])
+
+    truncator = KeepNTruncator(2)
+    
+    removed_count = obs.truncate(truncator)
+    
+    assert removed_count == 2
+    assert len(obs) == 2
+    assert obs[0] == PauliTerm("IIII", 0.5)
+    assert obs[1] == PauliTerm("YYYY", -0.5)
+
 def test_lambda_truncator_on_observable_def():
     """Tests truncation based on a custom Python function."""
     # This predicate removes any term containing a 'Y'