implement tests_data_path in atomic

tardis-sn · Jul 21, 2016 · 4a6fb5f · 4a6fb5f
1 parent ed4854e
commit 4a6fb5f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tardis/io/atomic.py b/tardis/io/atomic.py
@@ -20,11 +20,14 @@ class AtomDataNotPreparedError(Exception):
 
 logger = logging.getLogger(__name__)
 
+tardis_dir = os.path.dirname(os.path.realpath(tardis.__file__))
 
 def data_path(fname):
-    tardis_dir = os.path.dirname(os.path.realpath(tardis.__file__))
     return os.path.join(tardis_dir, 'data', fname)
 
+def tests_data_path(fname):
+    return os.path.join(tardis_dir, 'tests', 'data', fname)
+
 
 default_atom_h5_path = data_path('atom_data.h5')
 atomic_symbols_data = np.recfromtxt(data_path('atomic_symbols.dat'),

diff --git a/tardis/io/tests/test_atomic.py b/tardis/io/tests/test_atomic.py
@@ -11,7 +11,7 @@ def default_atom_h5_path():
 
 @pytest.fixture(scope="module")
 def chianti_he_db_h5_path():
-    return atomic.data_path('chianti_he_db.h5')
+    return atomic.tests_data_path('chianti_he_db.h5')
 
 
 def test_data_path():