Rename image class

TUW-GEO · Aug 2, 2023 · 2f1a45b · 2f1a45b
1 parent c5f44f2
commit 2f1a45b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/src/repurpose/stack.py b/src/repurpose/stack.py
@@ -18,7 +18,7 @@
 """
 
 
-class RegularImageStack:
+class Regular3dimImageStack:
 
  def __init__(self,
  grid,

diff --git a/src/repurpose/ts2img.py b/src/repurpose/ts2img.py
@@ -14,7 +14,7 @@
 from datetime import datetime
 import warnings
 
-from repurpose.image import RegularImageStack
+from repurpose.stack import Regular3dimImageStack
 
 """
 TODOs:
@@ -25,7 +25,7 @@
 """
 
 def _convert(converter: 'Ts2Img',
- writer: RegularImageStack,
+ writer: Regular3dimImageStack,
  img_gpis: np.ndarray,
  lons: np.ndarray,
  lats: np.ndarray) -> xr.Dataset:
@@ -139,7 +139,7 @@ def __init__(self, ts_reader, img_grid, timestamps,
  max_dist=18000, time_collocation=True):
 
  self.ts_reader = ts_reader
- self.img_grid: CellGrid = RegularImageStack._eval_grid(img_grid)
+ self.img_grid: CellGrid = Regular3dimImageStack._eval_grid(img_grid)
  self.timestamps = timestamps
 
  if variables is not None:
@@ -166,7 +166,7 @@ def _cell_writer(self, cell: int, timestamps_chunk: pd.DatetimeIndex):
  n_lons = np.unique(cellgrid.activearrlon).size
  cellgrid.shape = (n_lats, n_lons)
 
- writer = RegularImageStack(
+ writer = Regular3dimImageStack(
  cellgrid, timestamps=timestamps_chunk, zlib=False,
  time_collocation=self.time_collocation)
 

diff --git a/tests/test_image.py b/tests/test_image.py
@@ -2,7 +2,7 @@
 import unittest
 import os
 import pandas as pd
-from repurpose.image import RegularImageStack
+from repurpose.stack import Regular3dimImageStack
 import tempfile
 import pytest
 import xarray as xr
@@ -19,7 +19,7 @@ class TestRegularImageStackNoCollocation(unittest.TestCase):
  """
  def setUp(self) -> None:
  self.timestamps = pd.date_range('2020-01-01', '2020-01-02', freq='6H')
- self.writer = RegularImageStack.from_genreg(
+ self.writer = Regular3dimImageStack.from_genreg(
  resolution=0.25, extent=bbox_spain, timestamps=self.timestamps,
  time_collocation=False, reference_time="1900-01-01T00:00:00",
  zlib=True)
@@ -133,7 +133,7 @@ def setUp(self) -> None:
  grid_spain.shape = (np.unique(grid_spain.activearrlat).size,
  np.unique(grid_spain.activearrlon).size)
 
- self.writer = RegularImageStack(
+ self.writer = Regular3dimImageStack(
  grid=grid_spain, timestamps=self.img_timestamps, time_collocation=True,
  reference_time="1900-01-01T00:00:00", zlib=True)