Update unit tests

hiperwalk · Jun 11, 2024 · 7309018 · 7309018
1 parent 013c594
commit 7309018
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/tests/unit/coined_cycle.py b/tests/unit/coined_cycle.py
@@ -21,7 +21,7 @@ def test_clockwise_roundabout(self):
         self.qw.set_coin('I')
         self.qw.set_marked([])
 
-        init_state = self.qw.state([[1., (0, 1)]])
+        init_state = self.qw.state([[1, (0, 1)]])
 
         num_steps = self.num_vert
         final_state = self.qw.simulate((num_steps, num_steps + 1),
@@ -35,7 +35,7 @@ def test_anticlockwise_roundabout(self):
         self.qw.set_coin('I')
         self.qw.set_marked([])
 
-        init_state = self.qw.state([[1., (0, self.num_vert - 1)]])
+        init_state = self.qw.state([[1, (0, self.num_vert - 1)]])
 
         num_steps = self.num_vert
         final_state = self.qw.simulate((num_steps, num_steps + 1),
@@ -45,7 +45,7 @@ def test_anticlockwise_roundabout(self):
         self.assertTrue(np.all(init_state == final_state))
 
     def test_hadamard_evolution_operator(self):
-        init_state = self.qw.state([(1., (0, 1))])
+        init_state = self.qw.state([(1, (0, 1))])
 
         num_steps = 2*self.num_vert
         states = self.qw.simulate(num_steps + 1, init_state)
@@ -83,7 +83,7 @@ def test_hpc_evolution_operator_matches_nonhpc(self):
         hpw.set_hpc(None)
         states = self.qw.simulate(num_steps + 1, init_state)
 
-        hpw.set_hpc('cpu')
+        hpw.set_hpc(HPC)
         hpc_states = self.qw.simulate(num_steps + 1, init_state)
 
         self.assertTrue(np.allclose(states, hpc_states,

diff --git a/tests/unit/coined_line.py b/tests/unit/coined_line.py
@@ -22,7 +22,7 @@ def test_persistent_shift_right_state_transfer(self):
         self.qw.set_coin('I')
         self.qw.set_marked([])
 
-        init_state = self.qw.state([[1., (0, 1)]])
+        init_state = self.qw.state([[1, (0, 1)]])
 
         num_steps = self.num_vert - 1
         final_state = self.qw.simulate((num_steps, num_steps + 1),
@@ -42,7 +42,7 @@ def test_persistent_shift_left_state_transfer(self):
         self.qw.set_marked([])
 
         init_state = self.qw.state(
-            [[1., (self.num_vert - 1, self.num_vert - 2)]])
+            [[1, (self.num_vert - 1, self.num_vert - 2)]])
 
         num_steps = self.num_vert - 1
         final_state = self.qw.simulate((num_steps, num_steps + 1),
@@ -60,13 +60,13 @@ def test_hpc_default_evolution_operator(self):
 
         num_steps = self.num_vert // 2
         center = self.num_vert // 2
-        # changed from complex vector to float
-        entries = [[1., (0, 1)]]
+        entries = [[1, (center, center + 1)],
+                   [-1j, (center, center - 1)]]
         init_state = self.qw.state(entries)
 
         hpw.set_hpc(None)
         states = self.qw.simulate((num_steps + 1), init_state)
-        hpw.set_hpc('cpu')
+        hpw.set_hpc(HPC)
         hpc_states = self.qw.simulate((num_steps + 1), init_state)
 
         self.assertTrue(
@@ -134,9 +134,9 @@ def test_uniform_state(self):
         even_verts = np.arange(0, self.num_vert, 2)
         state = self.qw.uniform_state(vertices=even_verts,
                                       arcs=even_arcs)
-        state2 = [0. if (a % 2 == 1 and
+        state2 = [0 if (a % 2 == 1 and
                         self.qw._graph.arc(a)[0] % 2 == 1)
-                  else 1.
+                  else 1
                   for a in range(num_arcs)]
         state2 = state2 / np.sqrt(np.sum(state2))
         self.assertTrue(np.allclose(state, state2))