add tests

cherab/compass/equilibrium/test/test_compass_equilibrium.py

@@ -0,0 +1,18 @@
+import unittest
+
+from cherab.compass.equilibrium import COMPASSEquilibrium
+
+import xarray as xr
+
+class TesFiestaEquilibria(unittest.TestCase):
+
+
+    def test_mandatory_content(self):
+        data = xr.Dataset()
+
+        #leave out time coordinate
+
+        with self.assertRaises(KeyError):
+            COMPASSEquilibrium(data)
+
+

cherab/compass/equilibrium/test/test_efit.py

@@ -0,0 +1,13 @@
+import unittest
+from cherab.compass.equilibrium.efit import efit_from_cudb
+
+class TesFiestaEquilibria(unittest.TestCase):
+
+    def test_fiesta_from_cudb(self):
+        """Test initialisation of fiesta data for COMPASS Upgrade from cudb."""
+
+        shot_number = 15001
+        time = 1.1
+
+        equilibrium = efit_from_cudb(shot_number)
+        time_slice = equilibrium.time_slice(1.5)

cherab/compass/equilibrium/test/test_fiesta.py

@@ -0,0 +1,13 @@
+import unittest
+from cherab.compass.equilibrium.fiesta import fiesta_from_cudb
+
+class TesFiestaEquilibria(unittest.TestCase):
+
+    def test_fiesta_from_cudb(self):
+        """Test initialisation of COMPASS EFIT equilibrium from cdb."""
+
+        shot_number = 6400
+        time = 1.5
+
+        equilibrium = fiesta_from_cudb(shot_number)
+        time_slice = equilibrium.time_slice(1.5)