Skip to content

Commit

Permalink
sort imports and fix linting errors for tests folder
Browse files Browse the repository at this point in the history
  • Loading branch information
kevinsantana11 committed Jan 27, 2024
1 parent ef33dd4 commit 48acc1f
Show file tree
Hide file tree
Showing 9 changed files with 61 additions and 48 deletions.
7 changes: 4 additions & 3 deletions tests/datasets_tests.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
from clouddrift import datasets
from clouddrift.ragged import apply_ragged, subset
import numpy as np
import unittest

import numpy as np

from clouddrift import datasets
from clouddrift.ragged import apply_ragged, subset

if __name__ == "__main__":
unittest.main()
Expand Down
13 changes: 7 additions & 6 deletions tests/kinematics_tests.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,18 @@
import unittest

import numpy as np
import xarray as xr

from clouddrift.kinematics import (
inertial_oscillation_from_position,
kinetic_energy,
position_from_velocity,
residual_position_from_displacement,
velocity_from_position,
spin,
velocity_from_position,
)
from clouddrift.sphere import EARTH_RADIUS_METERS, coriolis_frequency
from clouddrift.raggedarray import RaggedArray
import unittest
import numpy as np
import xarray as xr

from clouddrift.sphere import EARTH_RADIUS_METERS, coriolis_frequency

if __name__ == "__main__":
unittest.main()
Expand Down
5 changes: 3 additions & 2 deletions tests/pairs_tests.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
from clouddrift import datasets, pairs, ragged, sphere
import unittest

import numpy as np
import pandas as pd
import unittest
import xarray as xr

from clouddrift import datasets, pairs, ragged, sphere

if __name__ == "__main__":
unittest.main()
Expand Down
9 changes: 5 additions & 4 deletions tests/plotting_tests.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,12 @@
import cartopy.crs as ccrs
import matplotlib.pyplot as plt
import numpy as np
import sys
import unittest
from unittest.mock import patch
from clouddrift.plotting import plot_ragged

import cartopy.crs as ccrs
import matplotlib.pyplot as plt
import numpy as np

from clouddrift.plotting import plot_ragged

if __name__ == "__main__":
unittest.main()
Expand Down
15 changes: 8 additions & 7 deletions tests/ragged_tests.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,11 @@
import unittest
from concurrent import futures
from datetime import datetime, timedelta

import numpy as np
import pandas as pd
import xarray as xr

from clouddrift.kinematics import velocity_from_position
from clouddrift.ragged import (
apply_ragged,
Expand All @@ -10,13 +18,6 @@
unpack,
)
from clouddrift.raggedarray import RaggedArray
import unittest
import numpy as np
import xarray as xr
import pandas as pd
from datetime import datetime, timedelta
from concurrent import futures


if __name__ == "__main__":
unittest.main()
Expand Down
8 changes: 5 additions & 3 deletions tests/raggedarray_tests.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,12 @@
import os
import unittest
from unittest import TestCase
import os
import xarray as xr

import awkward as ak
import numpy as np
import xarray as xr

from clouddrift import RaggedArray
import awkward as ak

NETCDF_ARCHIVE = "test_archive.nc"
PARQUET_ARCHIVE = "test_archive.parquet"
Expand Down
10 changes: 6 additions & 4 deletions tests/signal_tests.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,15 @@
import unittest

import numpy as np
import xarray as xr

from clouddrift.signal import (
analytic_signal,
cartesian_to_rotary,
rotary_to_cartesian,
ellipse_parameters,
modulated_ellipse_signal,
rotary_to_cartesian,
)
import numpy as np
import unittest
import xarray as xr

if __name__ == "__main__":
unittest.main()
Expand Down
26 changes: 14 additions & 12 deletions tests/sphere_tests.py
Original file line number Diff line number Diff line change
@@ -1,23 +1,25 @@
import unittest

import numpy as np
import xarray as xr

from clouddrift.sphere import (
EARTH_RADIUS_METERS,
bearing,
cartesian_to_spherical,
cartesian_to_tangentplane,
coriolis_frequency,
cumulative_distance,
distance,
plane_to_sphere,
position_from_distance_and_bearing,
recast_lon,
recast_lon180,
recast_lon360,
plane_to_sphere,
sphere_to_plane,
distance,
bearing,
position_from_distance_and_bearing,
spherical_to_cartesian,
cartesian_to_spherical,
tangentplane_to_cartesian,
cartesian_to_tangentplane,
coriolis_frequency,
cumulative_distance,
EARTH_RADIUS_METERS,
)
import unittest
import numpy as np
import xarray as xr

ONE_DEGREE_METERS = np.deg2rad(EARTH_RADIUS_METERS)

Expand Down
16 changes: 9 additions & 7 deletions tests/wavelet_tests.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,17 @@
import unittest

import numpy as np

from clouddrift.wavelet import (
morse_wavelet_transform,
wavelet_transform,
morse_wavelet,
morse_freq,
_morsehigh,
morse_amplitude,
morse_freq,
morse_logspace_freq,
morse_properties,
_morsehigh,
morse_wavelet,
morse_wavelet_transform,
wavelet_transform,
)
import numpy as np
import unittest

if __name__ == "__main__":
unittest.main()
Expand Down

0 comments on commit 48acc1f

Please sign in to comment.