Tests for 'devices'

.gitignore

@@ -5,3 +5,4 @@ __pycache__/
 env-pasqal-qutip/
 .DS_Store
 /*.ipynb
+*.coverage

pulser/devices.py

@@ -26,7 +26,7 @@ def __init__(self, qubits):
             raise TypeError("The qubits must be a in a dict or Register class "
                             "instance.")
 
-        self.check_array(list(register.qubits.values()))
+        self._check_array(list(register.qubits.values()))
         self._register = register
 
     @property
@@ -75,7 +75,7 @@ def qubits(self):
         """The dictionary of qubit names and their positions."""
         return self._register.qubits
 
-    def check_array(self, atoms):
+    def _check_array(self, atoms):
         if len(atoms) > self.max_atom_num:
             raise ValueError("Too many atoms in the array, accepts at most"
                              "{} atoms.".format(self.max_atom_num))

tests/test_channels.py

@@ -1,5 +1,4 @@
 import pytest
-import numpy as np
 
 from pulser.channels import Raman, Rydberg
 

tests/test_devices.py

@@ -0,0 +1,35 @@
+import pytest
+
+from pulser import Register
+from pulser.devices import Chadoq2
+
+
+def test_init():
+    with pytest.raises(TypeError):
+        Chadoq2([1, 2, 4])
+        Chadoq2('q1')
+        Chadoq2(('a', 'b', 'c'))
+
+    qubits = dict(enumerate([(0, 0), (10, 0)]))
+    dev1 = Chadoq2(qubits)
+    reg = Register(qubits)
+    dev2 = Chadoq2(reg)
+    assert dev1.qubits == dev2.qubits
+
+
+def test_check_array():
+    with pytest.raises(ValueError, match='Too many atoms'):
+        Chadoq2(Register.square(50))
+
+    coords = [(100, 0), (-100, 0)]
+    with pytest.raises(ValueError, match='at most 50um away from the center'):
+        Chadoq2(Register.from_coordinates(coords))
+
+    with pytest.raises(ValueError, match='must be 2D vectors'):
+        coords += [(-10, 4, 0)]
+        Chadoq2(dict(enumerate(coords)))
+
+    with pytest.raises(ValueError, match="don't respect the minimal distance"):
+        Chadoq2(Register.triangular_lattice(3, 4, spacing=3.9))
+
+    Chadoq2(Register.rectangle(5, 10, spacing=5))