Add standard packages to shorten docstring example

conftest.py

@@ -0,0 +1,19 @@
+# content of conftest.py
+import pytest
+import numpy
+import pandas
+import mikeio
+
+
+@pytest.fixture(autouse=True)
+def add_np(doctest_namespace):
+    doctest_namespace["np"] = numpy
+
+@pytest.fixture(autouse=True)
+def add_pd(doctest_namespace):
+    doctest_namespace["pd"] = pandas
+
+
+@pytest.fixture(autouse=True)
+def add_mikeio(doctest_namespace):
+    doctest_namespace["mikeio"] = mikeio

mikeio/dfsu.py

@@ -870,8 +870,6 @@ def write_header(
         Examples
         --------
         >>> from datetime import datetime
-        >>> import numpy as np
-        >>> import mikeio
         >>> meshfilename = "tests/testdata/north_sea_2.mesh"
         >>> outfilename = "bigfile.dfsu"
         >>> dfs = mikeio.Dfsu(meshfilename)
@@ -1226,7 +1224,6 @@ def extract_track(self, track, items=None, method="nearest", dtype=np.float32):
 
         Examples
         --------
-        >>> import mikeio
         >>> dfsu = mikeio.open("tests/testdata/NorthSea_HD_and_windspeed.dfsu")
         >>> ds = dfsu.extract_track("tests/testdata/altimetry_NorthSea_20171027.csv")
         >>> ds

mikeio/dfsu_spectral.py

@@ -113,7 +113,6 @@ def read(
 
         Examples
         --------
-        >>> import mikeio
         >>> mikeio.read("tests/testdata/line_spectra.dfsu")
         <mikeio.Dataset>
         dims: (time:4, node:10, direction:16, frequency:25)
@@ -293,7 +292,6 @@ def plot_spectrum(
 
         Examples
         --------
-        >>> import mikeio
         >>> dfs = mikeio.Dfsu("tests/testdata/area_spectra.dfsu")
         >>> ds = dfs.read(items="Energy density")
         >>> spectrum = ds[0][0, 0, :, :] # first timestep, element 0