Fusion Touch Ups + Fusion Verification

python/tests/test_anyon.py

@@ -1,10 +1,10 @@
 import pytest
-from anyon_braiding_simulator import Anyon, IsingTopoCharge
+from anyon_braiding_simulator import Anyon, IsingTopoCharge, TopoCharge
 
 
 @pytest.mark.anyon
 def test_anyon():
-    anyon = Anyon('thisisatest', IsingTopoCharge.Psi, (1, 1))
+    anyon = Anyon('thisisatest', TopoCharge.from_ising(IsingTopoCharge.Psi), (1, 1))
     assert anyon.name == 'thisisatest'
-    assert anyon.charge == IsingTopoCharge.Psi
+    assert anyon.charge.get_ising() == IsingTopoCharge.Psi
     assert anyon.position == (1, 1)

python/tests/test_braiding.py

@@ -10,16 +10,16 @@
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'model')))
 
-from anyon_braiding_simulator import Anyon, IsingTopoCharge
+from anyon_braiding_simulator import Anyon, IsingTopoCharge, TopoCharge
 
 
 @pytest.fixture
 def setup_anyons():
     return [
-        Anyon('A', IsingTopoCharge.Psi, (1, 1)),
-        Anyon('B', IsingTopoCharge.Psi, (2, 2)),
-        Anyon('C', IsingTopoCharge.Psi, (3, 3)),
-        Anyon('D', IsingTopoCharge.Psi, (4, 4)),
+        Anyon('A', TopoCharge.from_ising(IsingTopoCharge.Psi), (1, 1)),
+        Anyon('B', TopoCharge.from_ising(IsingTopoCharge.Psi), (2, 2)),
+        Anyon('C', TopoCharge.from_ising(IsingTopoCharge.Psi), (3, 3)),
+        Anyon('D', TopoCharge.from_ising(IsingTopoCharge.Psi), (4, 4)),
     ]
 
 
@@ -36,7 +36,7 @@ def test_braid_initialization(setup_anyons):
 
     # Test with duplicate anyon names
     duplicate_anyons = setup_anyons[:]
-    duplicate_anyons[3] = Anyon('A', IsingTopoCharge.Psi, (4, 4))
+    duplicate_anyons[3] = Anyon('A', TopoCharge.from_ising(IsingTopoCharge.Psi), (4, 4))
     with pytest.raises(ValueError, match='Duplicate anyon names detected'):
         Braid(duplicate_anyons)
 

python/tests/test_fusion.py

@@ -1,12 +1,12 @@
 import pytest
-from anyon_braiding_simulator import Anyon, Fusion, FusionPair, IsingTopoCharge, State
+from anyon_braiding_simulator import Anyon, AnyonModel, Fusion, FusionPair, IsingTopoCharge, State, TopoCharge
 
 
 @pytest.fixture
 def state() -> State:
     state = State()
     for i in range(6):
-        state.add_anyon(Anyon(f'{i}', IsingTopoCharge.Sigma, (0, 0)))
+        state.add_anyon(Anyon(f'{i}', TopoCharge.from_ising(IsingTopoCharge.Sigma), (0, 0)))
     return state
 
 
@@ -32,13 +32,13 @@ def test_apply_ising_fusion(state):
     vacuum = [0, 1, 0]
     sigma = [0, 0, 1]
 
-    assert fusion.apply_fusion(psi, psi) == vacuum
-    assert fusion.apply_fusion(vacuum, vacuum) == vacuum
-    assert fusion.apply_fusion(sigma, sigma) == [psi[i] + vacuum[i] for i in range(3)]
+    assert fusion.apply_fusion(psi, psi, AnyonModel.Ising) == vacuum
+    assert fusion.apply_fusion(vacuum, vacuum, AnyonModel.Ising) == vacuum
+    assert fusion.apply_fusion(sigma, sigma, AnyonModel.Ising) == [psi[i] + vacuum[i] for i in range(3)]
 
     psi_sigma = [1, 0, 1]
-    assert fusion.apply_fusion(psi_sigma, psi_sigma) == [1, 2, 2]
-    assert not fusion.apply_fusion(psi_sigma, psi_sigma) == [2, 1, 2]  # get owned rishi
+    assert fusion.apply_fusion(psi_sigma, psi_sigma, AnyonModel.Ising) == [1, 2, 2]
+    assert not fusion.apply_fusion(psi_sigma, psi_sigma, AnyonModel.Ising) == [2, 1, 2]  # get owned rishi
 
 
 @pytest.mark.fusion
@@ -52,4 +52,18 @@ def test_qubit_enc(state):
     fusion = Fusion(state)
     correct = [FusionPair(0, 1), FusionPair(2, 4), FusionPair(2, 3)]
 
-    assert set(map(str, fusion.qubit_enc())) == set(map(str, correct))
+    assert set(map(str, fusion.qubit_enc(AnyonModel.Ising))) == set(map(str, correct))
+
+
+@pytest.mark.fusion
+def test_verify_fusion_result(state):
+    fusion = Fusion(state)
+    assert not fusion.verify_fusion_result(TopoCharge.from_ising(IsingTopoCharge.Sigma), AnyonModel.Ising)
+    assert fusion.verify_fusion_result(TopoCharge.from_ising(IsingTopoCharge.Vacuum), AnyonModel.Ising)
+    assert fusion.verify_fusion_result(TopoCharge.from_ising(IsingTopoCharge.Psi), AnyonModel.Ising)
+
+    state.add_anyon(Anyon('7', TopoCharge.from_ising(IsingTopoCharge.Sigma), (0, 0)))
+    fusion = Fusion(state)
+    assert not fusion.verify_fusion_result(TopoCharge.from_ising(IsingTopoCharge.Vacuum), AnyonModel.Ising)
+    assert not fusion.verify_fusion_result(TopoCharge.from_ising(IsingTopoCharge.Psi), AnyonModel.Ising)
+    assert fusion.verify_fusion_result(TopoCharge.from_ising(IsingTopoCharge.Sigma), AnyonModel.Ising)

python/tests/test_state.py

@@ -1,5 +1,5 @@
 import pytest
-from anyon_braiding_simulator import Anyon, FusionPair, IsingTopoCharge, State
+from anyon_braiding_simulator import Anyon, FusionPair, IsingTopoCharge, State, TopoCharge
 
 
 @pytest.fixture
@@ -10,20 +10,20 @@ def state() -> State:
 @pytest.mark.state
 def test_add_anyon(state):
     for i in range(100):
-        anyon = Anyon(f'{i}', IsingTopoCharge.Sigma, (0, 0))
+        anyon = Anyon(f'{i}', TopoCharge.from_ising(IsingTopoCharge.Sigma), (0, 0))
         state.add_anyon(anyon)
 
     for anyon in state.anyons:
         assert anyon.name in [f'{i}' for i in range(100)]
-        assert anyon.charge == IsingTopoCharge.Sigma
+        assert anyon.charge.get_ising() == IsingTopoCharge.Sigma
         assert anyon.position == (0, 0)
 
     assert len(state.anyons) == 100
 
 
 def test_add_operation(state):
     for i in range(101):
-        anyon = Anyon(f'{i}', IsingTopoCharge.Sigma, (0, 0))
+        anyon = Anyon(f'{i}', TopoCharge.from_ising(IsingTopoCharge.Sigma), (0, 0))
         state.add_anyon(anyon)
 
     assert state.add_operation(1, FusionPair(0, 1))