util: better test_phys_fields.py

Author: paugier

fluidsim/util/test_phys_fields.py

@@ -1,19 +1,23 @@
+import pytest
+
 import numpy as np
 
 import h5py
 
 from fluidsim.util.phys_fields import (
     name_file_from_time_approx,
     compute_file_name,
+    time_from_path,
 )
 
 
-def test_name_file_from_time_approx(tmp_path):
-    dt = 0.0004
-    times = dt * np.arange(11)
+dt = 0.0004
+times = dt * np.arange(11)
+
 
-    path_dir = tmp_path / "run_dir"
-    path_dir.mkdir()
+@pytest.fixture(scope="module")
+def path_dir_with_files(tmp_path_factory):
+    path_dir = tmp_path_factory.mktemp("run_dir")
 
     str_width, ext = 7, "h5"
 
@@ -27,9 +31,20 @@ def test_name_file_from_time_approx(tmp_path):
             file.attrs["time"] = time
             file.attrs["it"] = it
 
-        # print(path_file)
+    return path_dir
+
+
+def test_time_from_path(path_dir_with_files):
+    paths = sorted(path_dir_with_files.glob("*.h5"))
+    for it, path in enumerate(paths):
+        t_from_path = time_from_path(path)
+        assert t_from_path == round(it * dt, 3), (path.name, t_from_path, it * dt)
+
+
+def test_name_file_from_time_approx(path_dir_with_files):
+    path_dir = path_dir_with_files
 
     name_file_last = name_file_from_time_approx(path_dir)
-    assert name_file_last == path_file.name
+    assert name_file_last == "state_phys_t000.004_it=10.h5"
 
     # TODO: add assert statements to get bugs!