Generative rotated molecules from a finite set of particules

compSPI · Jul 10, 2021 · 8dd5c80 · 8dd5c80
1 parent 576b45c
commit 8dd5c80
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sim_volumes.py b/sim_volumes.py
@@ -8,7 +8,7 @@
 PARTICULES = np.asarray([[1, 0, 0, 3], [0, 8, 0, 5], [0, 0, 7, 9]])
 N_VOLUMES = 2000
 VOL_SIZE = 64
-NAME = '4_points_3d'
+NAME = "4_points_3d"
 CENTER = 2
 
 

diff --git a/test_sim_volumes.py b/test_sim_volumes.py
@@ -11,7 +11,7 @@ class TestSimVolumes:
     def test_get_random_quat():
         quaternions = sim_volumes.get_random_quat(5)
         assert quaternions.shape == (5, 4)
-        assert type(quaternions) == np.ndarray
+        assert type(quaternions) is np.ndarray
 
     @staticmethod
     def test_uniform_rotations():