From 6417d27a33647a935db59ba66be1611e8ba7ed72 Mon Sep 17 00:00:00 2001 From: juli6nne Date: Sat, 22 Jun 2024 06:22:10 -0400 Subject: [PATCH] topocharge fix? --- python/tests/test_braiding.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/python/tests/test_braiding.py b/python/tests/test_braiding.py index 3c3bac8..bf54217 100644 --- a/python/tests/test_braiding.py +++ b/python/tests/test_braiding.py @@ -3,12 +3,11 @@ import sys import numpy as np -# Ensure correct path to import modules sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'anyon_braiding_simulator'))) from Braiding import Braid from Model import Model -from anyon_braiding_simulator import Anyon, AnyonModel, IsingTopoCharge, State, FusionPair +from anyon_braiding_simulator import Anyon, AnyonModel, IsingTopoCharge, TopoCharge, State, FusionPair @pytest.fixture @@ -16,10 +15,10 @@ def setup_state_and_anyons(): model = Model(AnyonModel.Ising) state = State() anyons = [ - 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)) ] for anyon in anyons: state.add_anyon(anyon) @@ -43,7 +42,7 @@ def test_braid_initialization(setup_state_and_anyons): # Test with duplicate anyon names state_duplicate = State() duplicate_anyons = anyons[:] - duplicate_anyons[3] = Anyon('A', IsingTopoCharge.Psi, (4, 4)) + duplicate_anyons[3] = Anyon('A', TopoCharge.from_ising(IsingTopoCharge.Psi), (4, 4)) for anyon in duplicate_anyons: state_duplicate.add_anyon(anyon) with pytest.raises(ValueError, match='Duplicate anyon names detected'): @@ -140,7 +139,7 @@ def setup_state(): # Initialize the state with 6 anyons 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))) # Add fusion operations state.add_operation(1, FusionPair(0, 1)) @@ -177,7 +176,7 @@ def test_swap_to_qubit(setup_state): @pytest.fixture def setup_braid(): state = State() - anyons = [Anyon(f'{i}', IsingTopoCharge.Sigma, (0,0)) for i in range(6)] + anyons = [Anyon(f'{i}', TopoCharge.from_ising(IsingTopoCharge.Sigma), (0,0)) for i in range(6)] for anyon in anyons: state.add_anyon(anyon) model = Model(AnyonModel.Ising)