From 992b8cc579f314af2cf60b09879028c025a3fc9b Mon Sep 17 00:00:00 2001
From: MatejTomes <tomes@ipp.cas.cz>
Date: Fri, 12 Jun 2020 10:54:54 +0200
Subject: [PATCH] add tests

---
 cherab/compass/equilibrium/test/__init__.py    |  0
 .../test/test_compass_equilibrium.py           | 18 ++++++++++++++++++
 cherab/compass/equilibrium/test/test_efit.py   | 13 +++++++++++++
 cherab/compass/equilibrium/test/test_fiesta.py | 13 +++++++++++++
 4 files changed, 44 insertions(+)
 create mode 100644 cherab/compass/equilibrium/test/__init__.py
 create mode 100644 cherab/compass/equilibrium/test/test_compass_equilibrium.py
 create mode 100644 cherab/compass/equilibrium/test/test_efit.py
 create mode 100644 cherab/compass/equilibrium/test/test_fiesta.py

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