apply isort

Author: MuellerSeb

examples/01_create.py

@@ -6,10 +6,10 @@
 on a field site.
 """
 
-import numpy as np
-import welltestpy as wtp
 import anaflow as ana
+import numpy as np
 
+import welltestpy as wtp
 
 ###############################################################################
 # Create the field-site and the campaign

examples/06_triangulate.py

@@ -10,7 +10,8 @@
 """
 
 import numpy as np
-from welltestpy.tools import triangulate, sym, plot_well_pos
+
+from welltestpy.tools import plot_well_pos, sym, triangulate
 
 dist_mat = np.zeros((4, 4), dtype=float)
 dist_mat[0, 1] = 3  # distance between well 0 and 1

src/welltestpy/__init__.py

@@ -58,8 +58,8 @@
     __version__ = "0.0.0.dev0"
 
 from .data.campaignlib import Campaign, FieldSite
-from .data.testslib import PumpingTest
 from .data.data_io import load_campaign
+from .data.testslib import PumpingTest
 
 __all__ = ["__version__"]
 __all__ += ["data", "estimate", "process", "tools"]

src/welltestpy/data/__init__.py

@@ -80,33 +80,29 @@
     load_obs
     load_well
 """
-from . import varlib, testslib, campaignlib, data_io
-
+from . import campaignlib, data_io, testslib, varlib
+from .campaignlib import Campaign, FieldSite
+from .data_io import (
+    load_campaign,
+    load_fieldsite,
+    load_obs,
+    load_test,
+    load_var,
+    load_well,
+)
+from .testslib import PumpingTest
 from .varlib import (
-    Variable,
-    TimeVar,
-    HeadVar,
-    TemporalVar,
     CoordinatesVar,
-    Observation,
-    StdyObs,
     DrawdownObs,
+    HeadVar,
+    Observation,
     StdyHeadObs,
+    StdyObs,
+    TemporalVar,
+    TimeVar,
+    Variable,
     Well,
 )
-from .testslib import PumpingTest
-from .campaignlib import (
-    FieldSite,
-    Campaign,
-)
-from .data_io import (
-    load_var,
-    load_obs,
-    load_well,
-    load_campaign,
-    load_fieldsite,
-    load_test,
-)
 
 __all__ = [
     "Variable",

src/welltestpy/data/campaignlib.py

@@ -12,7 +12,7 @@
 from copy import deepcopy as dcopy
 
 from ..tools import plotter
-from . import data_io, varlib, testslib
+from . import data_io, testslib, varlib
 
 __all__ = ["FieldSite", "Campaign"]
 

src/welltestpy/data/data_io.py

@@ -8,17 +8,19 @@
 
 .. autosummary::
 """
-import os
 import csv
+import numbers
+import os
 import shutil
-import zipfile
 import tempfile
-from io import TextIOWrapper as TxtIO, BytesIO as BytIO
-import numbers
+import zipfile
+from io import BytesIO as BytIO
+from io import TextIOWrapper as TxtIO
+
 import numpy as np
 from packaging.version import parse as version_parse
 
-from . import varlib, campaignlib, testslib
+from . import campaignlib, testslib, varlib
 
 try:
     from .._version import __version__

src/welltestpy/data/testslib.py

@@ -14,9 +14,9 @@
 
 import numpy as np
 
-from ..tools import plotter, diagnostic_plots
-from . import varlib, data_io
 from ..process import processlib
+from ..tools import diagnostic_plots, plotter
+from . import data_io, varlib
 
 __all__ = ["Test", "PumpingTest"]
 

src/welltestpy/data/varlib.py

@@ -19,13 +19,13 @@
    TimeSeries
    Well
 """
-from copy import deepcopy as dcopy
 import numbers
+from copy import deepcopy as dcopy
+
 import numpy as np
 
 from . import data_io
 
-
 __all__ = [
     "Variable",
     "TimeVar",

src/welltestpy/estimate/__init__.py

@@ -40,19 +40,18 @@
    SteadyPumping
 """
 from . import estimators, spotpylib, steady_lib, transient_lib
-
 from .estimators import (
-    ExtTheis3D,
     ExtTheis2D,
-    Neuman2004,
-    Theis,
-    ExtThiem3D,
+    ExtTheis3D,
     ExtThiem2D,
+    ExtThiem3D,
+    Neuman2004,
     Neuman2004Steady,
+    Theis,
     Thiem,
 )
-from .transient_lib import TransientPumping
 from .steady_lib import SteadyPumping
+from .transient_lib import TransientPumping
 
 __all__ = ["estimators", "spotpylib", "steady_lib", "transient_lib"]
 __all__ += [

src/welltestpy/estimate/estimators.py

@@ -20,7 +20,6 @@
 
 from . import steady_lib, transient_lib
 
-
 __all__ = [
     "ExtTheis3D",
     "ExtTheis2D",

src/welltestpy/estimate/spotpylib.py

@@ -15,7 +15,6 @@
 import numpy as np
 import spotpy
 
-
 __all__ = ["TypeCurve", "fast_rep"]
 
 

src/welltestpy/estimate/steady_lib.py

@@ -9,17 +9,17 @@
 .. autosummary::
    SteadyPumping
 """
-from copy import deepcopy as dcopy
 import os
 import time as timemodule
+from copy import deepcopy as dcopy
 
 import numpy as np
 import spotpy
 
 from ..data import testslib
 from ..process import processlib
-from . import spotpylib
 from ..tools import plotter
+from . import spotpylib
 
 __all__ = [
     "SteadyPumping",

src/welltestpy/estimate/transient_lib.py

@@ -9,18 +9,18 @@
 .. autosummary::
    TransientPumping
 """
-from copy import deepcopy as dcopy
 import os
 import time as timemodule
+from copy import deepcopy as dcopy
 
+import anaflow as ana
 import numpy as np
 import spotpy
-import anaflow as ana
 
 from ..data import testslib
 from ..process import processlib
-from . import spotpylib
 from ..tools import plotter
+from . import spotpylib
 
 __all__ = [
     "TransientPumping",

src/welltestpy/process/__init__.py

@@ -17,10 +17,10 @@
     smoothing_derivative
 """
 from .processlib import (
-    normpumptest,
     combinepumptest,
-    filterdrawdown,
     cooper_jacob_correction,
+    filterdrawdown,
+    normpumptest,
     smoothing_derivative,
 )
 

src/welltestpy/process/processlib.py

@@ -14,6 +14,7 @@
    smoothing_derivative
 """
 from copy import deepcopy as dcopy
+
 import numpy as np
 from scipy import signal
 

src/welltestpy/tools/__init__.py

@@ -27,23 +27,20 @@
    plotsensitivity
    diagnostic_plot_pump_test
 """
-from . import plotter, trilib, diagnostic_plots
-
-from .trilib import triangulate, sym
-
+from . import diagnostic_plots, plotter, trilib
+from .diagnostic_plots import diagnostic_plot_pump_test
 from .plotter import (
     campaign_plot,
+    campaign_well_plot,
     fadeline,
     plot_well_pos,
-    campaign_well_plot,
-    plotfit_transient,
     plotfit_steady,
+    plotfit_transient,
     plotparainteract,
     plotparatrace,
     plotsensitivity,
 )
-
-from .diagnostic_plots import diagnostic_plot_pump_test
+from .trilib import sym, triangulate
 
 __all__ = [
     "triangulate",

src/welltestpy/tools/diagnostic_plots.py

@@ -12,13 +12,12 @@
 # pylint: disable=C0103
 
 
+import matplotlib.pyplot as plt
 import numpy as np
 
 from ..process import processlib
 from . import plotter
 
-import matplotlib.pyplot as plt
-
 
 def diagnostic_plot_pump_test(
     observation,

src/welltestpy/tools/plotter.py

@@ -20,15 +20,14 @@
 """
 # pylint: disable=C0103
 import copy
-import warnings
 import functools as ft
+import warnings
 
-import numpy as np
-from scipy.stats import gaussian_kde
-
-from mpl_toolkits.mplot3d import Axes3D
 import matplotlib.pyplot as plt
 import matplotlib.ticker as ticker
+import numpy as np
+from mpl_toolkits.mplot3d import Axes3D
+from scipy.stats import gaussian_kde
 
 
 def _get_fig_ax(

src/welltestpy/tools/trilib.py

@@ -12,8 +12,8 @@
 """
 # pylint: disable=C0103
 from copy import deepcopy as dcopy
-import numpy as np
 
+import numpy as np
 
 __all__ = ["triangulate", "sym"]
 

tests/test_process.py

@@ -2,9 +2,11 @@
 """
 This is the unittest of welltestpy.process.
 """
+import copy
 import unittest
+
 import numpy as np
-import copy
+
 import welltestpy as wtp
 
 

tests/test_welltestpy.py

@@ -4,16 +4,17 @@
 """
 
 import unittest
-import numpy as np
+
 import matplotlib as mpl
+import numpy as np
 
 mpl.use("Agg")
 
-import welltestpy as wtp
-from welltestpy.tools import triangulate, sym, plot_well_pos
-
 import anaflow as ana
 
+import welltestpy as wtp
+from welltestpy.tools import plot_well_pos, sym, triangulate
+
 
 class TestWTP(unittest.TestCase):
     def setUp(self):