Dataset with 1 and 2 timesteps == equidistant

Author: ecomodeller

mikeio/dutil.py

@@ -318,4 +318,7 @@ def _ipython_key_completions_(self):
 
     @property
     def is_equidistant(self):
+        if len(self.time) < 3:
+            return True
+
         return self.time.freq is not None

tests/test_dfs2.py

@@ -72,7 +72,11 @@ def test_non_equidistant_calendar(tmpdir):
 
     data = []
 
-    datetimes = [datetime.datetime(2012, 1, 1), datetime.datetime(2012, 2, 1)]
+    datetimes = [
+        datetime.datetime(2012, 1, 1),
+        datetime.datetime(2012, 2, 1),
+        datetime.datetime(2012, 2, 10),
+    ]
 
     nt = len(datetimes)
     nx = 20

tests/test_dfsu.py

@@ -524,6 +524,26 @@ def test_write_from_dfsu(tmpdir):
     assert dfs.end_time == newdfs.end_time
 
 
+def test_write_from_dfsu_2_time_steps(tmpdir):
+
+    sourcefilename = os.path.join("tests", "testdata", "HD2D.dfsu")
+    outfilename = os.path.join(tmpdir.dirname, "simple.dfsu")
+    dfs = Dfsu(sourcefilename)
+
+    ds = dfs.read(time_steps=[0, 1])
+
+    assert ds.is_equidistant  # Data with two time steps is per definition equidistant
+
+    dfs.write(outfilename, ds)
+
+    assert os.path.exists(outfilename)
+
+    newdfs = Dfsu(outfilename)
+    assert dfs.start_time == newdfs.start_time
+    assert dfs.timestep == newdfs.timestep
+    assert dfs.end_time != newdfs.end_time
+
+
 def test_write_from_dfsu_default_items_numbered(tmpdir):
 
     sourcefilename = os.path.join("tests", "testdata", "HD2D.dfsu")