change module name to polytope-feature

ecmwf · Oct 7, 2024 · 3954c0c · 3954c0c
1 parent 6400ce4
commit 3954c0c
Show file tree

Hide file tree

Showing 101 changed files with 199 additions and 196 deletions.
diff --git a/polytope/__init__.py → polytope_feature/__init__.py b/polytope/__init__.py → polytope_feature/__init__.py
diff --git a/polytope/datacube/__init__.py → polytope_feature/datacube/__init__.py b/polytope/datacube/__init__.py → polytope_feature/datacube/__init__.py
diff --git a/polytope/datacube/backends/__init__.py → ...ope_feature/datacube/backends/__init__.py b/polytope/datacube/backends/__init__.py → ...ope_feature/datacube/backends/__init__.py
diff --git a/polytope/datacube/backends/datacube.py → ...ope_feature/datacube/backends/datacube.py b/polytope/datacube/backends/datacube.py → ...ope_feature/datacube/backends/datacube.py
diff --git a/polytope/datacube/backends/fdb.py → polytope_feature/datacube/backends/fdb.py b/polytope/datacube/backends/fdb.py → polytope_feature/datacube/backends/fdb.py
diff --git a/polytope/datacube/backends/mock.py → polytope_feature/datacube/backends/mock.py b/polytope/datacube/backends/mock.py → polytope_feature/datacube/backends/mock.py
diff --git a/polytope/datacube/backends/xarray.py → polytope_feature/datacube/backends/xarray.py b/polytope/datacube/backends/xarray.py → polytope_feature/datacube/backends/xarray.py
diff --git a/polytope/datacube/datacube_axis.py → polytope_feature/datacube/datacube_axis.py b/polytope/datacube/datacube_axis.py → polytope_feature/datacube/datacube_axis.py
diff --git a/polytope/datacube/index_tree.proto → polytope_feature/datacube/index_tree.proto b/polytope/datacube/index_tree.proto → polytope_feature/datacube/index_tree.proto
diff --git a/polytope/datacube/index_tree_pb2.py → polytope_feature/datacube/index_tree_pb2.py b/polytope/datacube/index_tree_pb2.py → polytope_feature/datacube/index_tree_pb2.py
diff --git a/polytope/datacube/tensor_index_tree.py → ...ope_feature/datacube/tensor_index_tree.py b/polytope/datacube/tensor_index_tree.py → ...ope_feature/datacube/tensor_index_tree.py
diff --git a/...tope/datacube/transformations/__init__.py → ...ture/datacube/transformations/__init__.py b/...tope/datacube/transformations/__init__.py → ...ture/datacube/transformations/__init__.py
diff --git a/...ansformations/datacube_cyclic/__init__.py → ...ansformations/datacube_cyclic/__init__.py b/...ansformations/datacube_cyclic/__init__.py → ...ansformations/datacube_cyclic/__init__.py
diff --git a/...ations/datacube_cyclic/datacube_cyclic.py → ...ations/datacube_cyclic/datacube_cyclic.py b/...ations/datacube_cyclic/datacube_cyclic.py → ...ations/datacube_cyclic/datacube_cyclic.py
diff --git a/...nsformations/datacube_mappers/__init__.py → ...nsformations/datacube_mappers/__init__.py b/...nsformations/datacube_mappers/__init__.py → ...nsformations/datacube_mappers/__init__.py
diff --git a/...ions/datacube_mappers/datacube_mappers.py → ...ions/datacube_mappers/datacube_mappers.py b/...ions/datacube_mappers/datacube_mappers.py → ...ions/datacube_mappers/datacube_mappers.py
@@ -27,7 +27,7 @@ def __init__(self, name, mapper_options):
 
     def generate_final_transformation(self):
         map_type = _type_to_datacube_mapper_lookup[self.grid_type]
-        module = import_module("polytope.datacube.transformations.datacube_mappers.mapper_types." + self.grid_type)
+        module = import_module("polytope_feature.datacube.transformations.datacube_mappers.mapper_types." + self.grid_type)
         constructor = getattr(module, map_type)
         transformation = deepcopy(
             constructor(self.old_axis, self.grid_axes, self.grid_resolution, self.local_area, self._axis_reversed)

diff --git a/...datacube_mappers/mapper_types/__init__.py → ...datacube_mappers/mapper_types/__init__.py b/...datacube_mappers/mapper_types/__init__.py → ...datacube_mappers/mapper_types/__init__.py
diff --git a/.../datacube_mappers/mapper_types/healpix.py → .../datacube_mappers/mapper_types/healpix.py b/.../datacube_mappers/mapper_types/healpix.py → .../datacube_mappers/mapper_types/healpix.py
diff --git a/...be_mappers/mapper_types/healpix_nested.py → ...be_mappers/mapper_types/healpix_nested.py b/...be_mappers/mapper_types/healpix_nested.py → ...be_mappers/mapper_types/healpix_nested.py
diff --git a/...ube_mappers/mapper_types/local_regular.py → ...ube_mappers/mapper_types/local_regular.py b/...ube_mappers/mapper_types/local_regular.py → ...ube_mappers/mapper_types/local_regular.py
diff --git a/...tacube_mappers/mapper_types/octahedral.py → ...tacube_mappers/mapper_types/octahedral.py b/...tacube_mappers/mapper_types/octahedral.py → ...tacube_mappers/mapper_types/octahedral.py
diff --git a/...tacube_mappers/mapper_types/reduced_ll.py → ...tacube_mappers/mapper_types/reduced_ll.py b/...tacube_mappers/mapper_types/reduced_ll.py → ...tacube_mappers/mapper_types/reduced_ll.py
diff --git a/.../datacube_mappers/mapper_types/regular.py → .../datacube_mappers/mapper_types/regular.py b/.../datacube_mappers/mapper_types/regular.py → .../datacube_mappers/mapper_types/regular.py
diff --git a/...ansformations/datacube_merger/__init__.py → ...ansformations/datacube_merger/__init__.py b/...ansformations/datacube_merger/__init__.py → ...ansformations/datacube_merger/__init__.py
diff --git a/...ations/datacube_merger/datacube_merger.py → ...ations/datacube_merger/datacube_merger.py b/...ations/datacube_merger/datacube_merger.py → ...ations/datacube_merger/datacube_merger.py
diff --git a/...nsformations/datacube_reverse/__init__.py → ...nsformations/datacube_reverse/__init__.py b/...nsformations/datacube_reverse/__init__.py → ...nsformations/datacube_reverse/__init__.py
diff --git a/...ions/datacube_reverse/datacube_reverse.py → ...ions/datacube_reverse/datacube_reverse.py b/...ions/datacube_reverse/datacube_reverse.py → ...ions/datacube_reverse/datacube_reverse.py
diff --git a/...ansformations/datacube_transformations.py → ...ansformations/datacube_transformations.py b/...ansformations/datacube_transformations.py → ...ansformations/datacube_transformations.py
@@ -12,7 +12,7 @@ def create_transform(name, transformation_type_key, transformation_options):
         transformation_type = _type_to_datacube_transformation_lookup[transformation_type_key]
         transformation_file_name = _type_to_transformation_file_lookup[transformation_type_key]
         file_name = ".datacube_" + transformation_file_name
-        module = import_module("polytope.datacube.transformations" + file_name + file_name)
+        module = import_module("polytope_feature.datacube.transformations" + file_name + file_name)
         constructor = getattr(module, transformation_type)
         transformation_type_option = transformation_options
         new_transformation = deepcopy(constructor(name, transformation_type_option))

diff --git a/...rmations/datacube_type_change/__init__.py → ...rmations/datacube_type_change/__init__.py b/...rmations/datacube_type_change/__init__.py → ...rmations/datacube_type_change/__init__.py
diff --git a/...acube_type_change/datacube_type_change.py → ...acube_type_change/datacube_type_change.py b/...acube_type_change/datacube_type_change.py → ...acube_type_change/datacube_type_change.py
@@ -15,7 +15,7 @@ def __init__(self, name, type_options):
 
     def generate_final_transformation(self):
         map_type = _type_to_datacube_type_change_lookup[self.new_type]
-        module = import_module("polytope.datacube.transformations.datacube_type_change.datacube_type_change")
+        module = import_module("polytope_feature.datacube.transformations.datacube_type_change.datacube_type_change")
         constructor = getattr(module, map_type)
         transformation = deepcopy(constructor(self.name, self.new_type))
         return transformation

diff --git a/polytope/datacube/tree_encoding.py → polytope_feature/datacube/tree_encoding.py b/polytope/datacube/tree_encoding.py → polytope_feature/datacube/tree_encoding.py
diff --git a/polytope/engine/__init__.py → polytope_feature/engine/__init__.py b/polytope/engine/__init__.py → polytope_feature/engine/__init__.py
diff --git a/polytope/engine/engine.py → polytope_feature/engine/engine.py b/polytope/engine/engine.py → polytope_feature/engine/engine.py
diff --git a/polytope/engine/hullslicer.py → polytope_feature/engine/hullslicer.py b/polytope/engine/hullslicer.py → polytope_feature/engine/hullslicer.py
diff --git a/polytope/options.py → polytope_feature/options.py b/polytope/options.py → polytope_feature/options.py
diff --git a/polytope/polytope.py → polytope_feature/polytope.py b/polytope/polytope.py → polytope_feature/polytope.py
diff --git a/polytope/shapes.py → polytope_feature/shapes.py b/polytope/shapes.py → polytope_feature/shapes.py
diff --git a/polytope/utility/__init__.py → polytope_feature/utility/__init__.py b/polytope/utility/__init__.py → polytope_feature/utility/__init__.py
diff --git a/polytope/utility/combinatorics.py → polytope_feature/utility/combinatorics.py b/polytope/utility/combinatorics.py → polytope_feature/utility/combinatorics.py
diff --git a/polytope/utility/exceptions.py → polytope_feature/utility/exceptions.py b/polytope/utility/exceptions.py → polytope_feature/utility/exceptions.py
diff --git a/polytope/utility/geometry.py → polytope_feature/utility/geometry.py b/polytope/utility/geometry.py → polytope_feature/utility/geometry.py
diff --git a/polytope/utility/list_tools.py → polytope_feature/utility/list_tools.py b/polytope/utility/list_tools.py → polytope_feature/utility/list_tools.py
diff --git a/polytope/utility/profiling.py → polytope_feature/utility/profiling.py b/polytope/utility/profiling.py → polytope_feature/utility/profiling.py
diff --git a/polytope/version.py → polytope_feature/version.py b/polytope/version.py → polytope_feature/version.py
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 __version__ = re.search(
     r'__version__\s*=\s*[\'"]([^\'"]*)[\'"]',
-    io.open("polytope/version.py", encoding="utf_8_sig").read(),
+    io.open("polytope_feature/version.py", encoding="utf_8_sig").read(),
 ).group(1)
 
 with open("requirements.txt") as f:

diff --git a/tests/test_axis_mappers.py b/tests/test_axis_mappers.py
@@ -1,13 +1,13 @@
 import pandas as pd
 
-from polytope.datacube.datacube_axis import (
+from polytope_feature.datacube.datacube_axis import (
     DatacubeAxisCyclic,
     FloatDatacubeAxis,
     IntDatacubeAxis,
     PandasTimedeltaDatacubeAxis,
     PandasTimestampDatacubeAxis,
 )
-from polytope.options import PolytopeOptions
+from polytope_feature.options import PolytopeOptions
 
 
 class TestAxisMappers:

diff --git a/tests/test_combinatorics.py b/tests/test_combinatorics.py
@@ -1,9 +1,9 @@
 import numpy as np
 import pytest
 
-from polytope import ConvexPolytope
-from polytope.utility.combinatorics import group, tensor_product, validate_axes
-from polytope.utility.exceptions import (
+from polytope_feature import ConvexPolytope
+from polytope_feature.utility.combinatorics import group, tensor_product, validate_axes
+from polytope_feature.utility.exceptions import (
     AxisNotFoundError,
     AxisOverdefinedError,
     AxisUnderdefinedError,

diff --git a/tests/test_cyclic_axis_over_negative_vals.py b/tests/test_cyclic_axis_over_negative_vals.py
@@ -2,9 +2,9 @@
 import pandas as pd
 import xarray as xr
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select
 
 
 class TestSlicingCyclicAxisNegVals:

diff --git a/tests/test_cyclic_axis_slicer_not_0.py b/tests/test_cyclic_axis_slicer_not_0.py
@@ -2,9 +2,9 @@
 import pandas as pd
 import xarray as xr
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select
 
 
 class TestSlicingCyclicAxisNotOverZero:

diff --git a/tests/test_cyclic_axis_slicing.py b/tests/test_cyclic_axis_slicing.py
@@ -2,9 +2,9 @@
 import pandas as pd
 import xarray as xr
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select
 
 
 class TestSlicingCyclic:

diff --git a/tests/test_cyclic_nearest.py b/tests/test_cyclic_nearest.py
@@ -3,9 +3,9 @@
 from eccodes import codes_grib_find_nearest, codes_grib_new_from_file
 from helper_functions import download_test_data
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Point, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Point, Select
 
 
 class TestRegularGrid:

diff --git a/tests/test_cyclic_simple.py b/tests/test_cyclic_simple.py
@@ -2,9 +2,9 @@
 import pandas as pd
 import xarray as xr
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select
 
 
 class TestSlicing3DXarrayDatacube:

diff --git a/tests/test_cyclic_snapping.py b/tests/test_cyclic_snapping.py
@@ -1,8 +1,8 @@
 import xarray as xr
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Select
 
 
 class TestSlicing3DXarrayDatacube:

diff --git a/tests/test_datacube_axes_init.py b/tests/test_datacube_axes_init.py
@@ -2,10 +2,10 @@
 from earthkit import data
 from helper_functions import download_test_data
 
-from polytope.datacube.datacube_axis import FloatDatacubeAxis
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select
+from polytope_feature.datacube.datacube_axis import FloatDatacubeAxis
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select
 
 
 class TestInitDatacubeAxes:

diff --git a/tests/test_datacube_mock.py b/tests/test_datacube_mock.py
@@ -1,7 +1,7 @@
 import pytest
 
-from polytope.datacube.backends.mock import MockDatacube
-from polytope.utility.exceptions import AxisNotFoundError, AxisOverdefinedError
+from polytope_feature.datacube.backends.mock import MockDatacube
+from polytope_feature.utility.exceptions import AxisNotFoundError, AxisOverdefinedError
 
 
 class TestMockDatacube:

diff --git a/tests/test_datacube_xarray.py b/tests/test_datacube_xarray.py
@@ -5,14 +5,14 @@
 import pytest
 import xarray as xr
 
-from polytope.datacube import Datacube, DatacubePath
-from polytope.datacube.backends.xarray import XArrayDatacube
-from polytope.datacube.datacube_axis import (
+from polytope_feature.datacube import Datacube, DatacubePath
+from polytope_feature.datacube.backends.xarray import XArrayDatacube
+from polytope_feature.datacube.datacube_axis import (
     FloatDatacubeAxis,
     IntDatacubeAxis,
     PandasTimestampDatacubeAxis,
 )
-from polytope.utility.exceptions import AxisNotFoundError, AxisOverdefinedError
+from polytope_feature.utility.exceptions import AxisNotFoundError, AxisOverdefinedError
 
 
 class TestXarrayDatacube:

diff --git a/tests/test_ecmwf_oper_data_fdb.py b/tests/test_ecmwf_oper_data_fdb.py
@@ -1,9 +1,9 @@
 import pandas as pd
 import pytest
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Point, Select, Span
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Point, Select, Span
 
 
 class TestSlicingFDBDatacube:

diff --git a/tests/test_engine_slicer.py b/tests/test_engine_slicer.py
@@ -1,6 +1,6 @@
-from polytope.datacube.backends.mock import MockDatacube
-from polytope.engine.hullslicer import HullSlicer
-from polytope.shapes import Box, Polygon
+from polytope_feature.datacube.backends.mock import MockDatacube
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.shapes import Box, Polygon
 
 
 class TestEngineSlicer:

diff --git a/tests/test_fdb_datacube.py b/tests/test_fdb_datacube.py
@@ -1,9 +1,9 @@
 import pandas as pd
 import pytest
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select, Span
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select, Span
 
 # import geopandas as gpd
 # import matplotlib.pyplot as plt

diff --git a/tests/test_fdb_unmap_tree.py b/tests/test_fdb_unmap_tree.py
@@ -1,9 +1,9 @@
 import pandas as pd
 import pytest
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select
 
 
 class TestSlicingFDBDatacube:

diff --git a/tests/test_float_type.py b/tests/test_float_type.py
@@ -2,9 +2,9 @@
 import pytest
 import xarray as xr
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Select, Span
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Select, Span
 
 
 class TestFloatType:

diff --git a/tests/test_healpix_mapper.py b/tests/test_healpix_mapper.py
@@ -2,12 +2,12 @@
 from earthkit import data
 from helper_functions import download_test_data, find_nearest_latlon
 
-from polytope.datacube.transformations.datacube_mappers.mapper_types.healpix import (
+from polytope_feature.datacube.transformations.datacube_mappers.mapper_types.healpix import (
     HealpixGridMapper,
 )
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select
 
 
 class TestHealpixGrid:

diff --git a/tests/test_healpix_nested_grid.py b/tests/test_healpix_nested_grid.py
@@ -3,12 +3,12 @@
 import pandas as pd
 import pytest
 
-from polytope.datacube.transformations.datacube_mappers.mapper_types.healpix_nested import (
+from polytope_feature.datacube.transformations.datacube_mappers.mapper_types.healpix_nested import (
     NestedHealpixGridMapper,
 )
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Box, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Box, Select
 
 
 class TestHealpixNestedGrid:

diff --git a/tests/test_hull_slicer.py b/tests/test_hull_slicer.py
@@ -2,15 +2,15 @@
 
 import pytest
 
-import polytope.engine.hullslicer
-from polytope import ConvexPolytope
-from polytope.datacube.backends.mock import MockDatacube
-from polytope.utility.profiling import benchmark
+from polytope_feature import ConvexPolytope
+from polytope_feature.datacube.backends.mock import MockDatacube
+from polytope_feature.engine.hullslicer import HullSlicer, slice
+from polytope_feature.utility.profiling import benchmark
 
 
 class TestHullSlicer:
     def setup_method(self, method):
-        self.slicer = polytope.engine.hullslicer.HullSlicer()
+        self.slicer = HullSlicer()
 
     def construct_nd_cube(self, dimension, lower=-1, upper=1):
         axes = [str(chr(97 + ax)) for ax in range(dimension)]
@@ -20,40 +20,40 @@ def construct_nd_cube(self, dimension, lower=-1, upper=1):
     def test_3D(self):
         p3 = self.construct_nd_cube(3)
         print(p3)
-        p2 = polytope.engine.hullslicer.slice(p3, "c", 0.5, 2)
+        p2 = slice(p3, "c", 0.5, 2)
         print(p2)
-        p1 = polytope.engine.hullslicer.slice(p2, "b", 0.5, 1)
+        p1 = slice(p2, "b", 0.5, 1)
         print(p1)
 
     @pytest.mark.skip(reason="This is too slow.")
     def test_4D(self):
         p = self.construct_nd_cube(4)
         print(p)
         while len(p.axes()) > 1:
-            p = polytope.engine.hullslicer.slice(p, p._axes[-1], 0.5, -1)
+            p = slice(p, p._axes[-1], 0.5, -1)
             print(p)
 
     @pytest.mark.skip(reason="This is too slow.")
     def test_ND(self):
         with benchmark("4D"):
             p = self.construct_nd_cube(4)
             while len(p.axes()) > 1:
-                p = polytope.engine.hullslicer.slice(p, p._axes[-1], 0.5, -1)
+                p = slice(p, p._axes[-1], 0.5, -1)
 
         with benchmark("5D"):
             p = self.construct_nd_cube(5)
             while len(p.axes()) > 1:
-                p = polytope.engine.hullslicer.slice(p, p._axes[-1], 0.5, -1)
+                p = slice(p, p._axes[-1], 0.5, -1)
 
         with benchmark("6D"):
             p = self.construct_nd_cube(6)
             while len(p.axes()) > 1:
-                p = polytope.engine.hullslicer.slice(p, p._axes[-1], 0.5, -1)
+                p = slice(p, p._axes[-1], 0.5, -1)
 
         with benchmark("7D"):
             p = self.construct_nd_cube(7)
             while len(p.axes()) > 1:
-                p = polytope.engine.hullslicer.slice(p, p._axes[-1], 0.5, -1)
+                p = slice(p, p._axes[-1], 0.5, -1)
 
         # QHull is not performant above 7D as per its documentation
         # with benchmark("8D"):

diff --git a/tests/test_hullslicer_engine.py b/tests/test_hullslicer_engine.py
@@ -1,11 +1,11 @@
 import numpy as np
 import xarray as xr
 
-from polytope.datacube.backends.xarray import XArrayDatacube
-from polytope.datacube.tensor_index_tree import TensorIndexTree
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope
-from polytope.shapes import Box
+from polytope_feature.datacube.backends.xarray import XArrayDatacube
+from polytope_feature.datacube.tensor_index_tree import TensorIndexTree
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope
+from polytope_feature.shapes import Box
 
 
 class TestSlicerComponents:

diff --git a/tests/test_incomplete_tree_fdb.py b/tests/test_incomplete_tree_fdb.py
@@ -2,9 +2,9 @@
 import pytest
 from helper_functions import download_test_data
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Select
 
 
 class TestRegularGrid:

diff --git a/tests/test_local_grid_cyclic.py b/tests/test_local_grid_cyclic.py
@@ -1,9 +1,9 @@
 import pandas as pd
 import pytest
 
-from polytope.engine.hullslicer import HullSlicer
-from polytope.polytope import Polytope, Request
-from polytope.shapes import Point, Select
+from polytope_feature.engine.hullslicer import HullSlicer
+from polytope_feature.polytope import Polytope, Request
+from polytope_feature.shapes import Point, Select
 
 
 class TestSlicingFDBDatacube: