22from unittest .mock import Mock , MagicMock , patch
33from dataclasses import FrozenInstanceError
44from gscrib import GCodeBuilder
5- from vpype_gscrib .heightmaps import FlatHeightMap
6- from vpype_gscrib .heightmaps import SparseHeightMap
7- from vpype_gscrib .heightmaps import RasterHeightMap
5+ from gscrib .enums import LengthUnits
6+ from gscrib .heightmaps import BaseHeightMap
7+ from gscrib .heightmaps import FlatHeightMap
8+ from gscrib .heightmaps import SparseHeightMap
9+ from gscrib .heightmaps import RasterHeightMap
810from vpype_gscrib .renderer import GContext
911from vpype_gscrib .config import RenderConfig
10- from gscrib .enums import LengthUnits
11- from vpype_gscrib .heightmaps import BaseHeightMap
1212
1313
1414# --------------------------------------------------------------------
@@ -93,7 +93,7 @@ def test_height_map_no_path(mock_gbuilder, render_config):
9393 context = GContext (mock_gbuilder , render_config )
9494 assert isinstance (context ._height_map , FlatHeightMap )
9595
96- @patch ("vpype_gscrib .heightmaps.SparseHeightMap.from_path" )
96+ @patch ("gscrib .heightmaps.SparseHeightMap.from_path" )
9797def test_build_height_map_sparse (from_path , mock_gbuilder , render_config ):
9898 heightmap = MagicMock (spec = SparseHeightMap )
9999 from_path .return_value = heightmap
@@ -105,7 +105,7 @@ def test_build_height_map_sparse(from_path, mock_gbuilder, render_config):
105105 heightmap .set_tolerance .assert_called_with (tolerance )
106106 heightmap .set_scale .assert_called_with (scale )
107107
108- @patch ("vpype_gscrib .heightmaps.RasterHeightMap.from_path" )
108+ @patch ("gscrib .heightmaps.RasterHeightMap.from_path" )
109109def test_build_height_map_raster (from_path , mock_gbuilder , render_config ):
110110 heightmap = MagicMock (spec = RasterHeightMap )
111111 from_path .return_value = heightmap
0 commit comments