diff --git a/cherab/compass/equilibrium/test/__init__.py b/cherab/compass/equilibrium/test/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/cherab/compass/equilibrium/test/test_compass_equilibrium.py b/cherab/compass/equilibrium/test/test_compass_equilibrium.py
new file mode 100644
index 0000000..c10ec8d
--- /dev/null
+++ b/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)
+
+
diff --git a/cherab/compass/equilibrium/test/test_efit.py b/cherab/compass/equilibrium/test/test_efit.py
new file mode 100644
index 0000000..586e4f9
--- /dev/null
+++ b/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)
diff --git a/cherab/compass/equilibrium/test/test_fiesta.py b/cherab/compass/equilibrium/test/test_fiesta.py
new file mode 100644
index 0000000..8a16838
--- /dev/null
+++ b/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)
\ No newline at end of file