Coverage: Add tests for get_current_rotation

pyxem · Feb 3, 2024 · 44c4673 · 44c4673
1 parent 8f24123
commit 44c4673
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/diffsims/tests/simulations/test_simulation.py b/diffsims/tests/simulations/test_simulation.py
@@ -184,6 +184,10 @@ def multi_simulation(self, al_phase):
         )
         return sim
 
+    def test_get_current_rotation(self, multi_simulation):
+        rot = multi_simulation.get_current_rotation()
+        np.testing.assert_array_equal(rot, multi_simulation.rotations[0].to_matrix()[0])
+
     def test_init(self, multi_simulation):
         assert isinstance(multi_simulation, Simulation2D)
         assert isinstance(multi_simulation.phases, Phase)