test update

Author: jtsodapop

labelbox/data/annotation_types/data/tiled_image.py

@@ -8,7 +8,7 @@
 import requests
 import numpy as np
 
-from retry import retry
+from retry import retry  #TODO not part of the package atm. need to add in?
 import tensorflow as f
 from PIL import Image
 from pyproj import Transformer

tests/data/annotation_types/test_tiled_image.py

@@ -2,8 +2,10 @@
 from labelbox.data.annotation_types.data import tiled_image
 from labelbox.data.annotation_types.geometry.point import Point
 from labelbox.data.annotation_types.data.tiled_image import (EPSG, TiledBounds,
-                                                             TileLayer)
+                                                             TileLayer,
+                                                             TiledImageData)
 from pydantic import ValidationError
+from retry import retry
 
 
 @pytest.mark.parametrize("epsg", list(EPSG))
@@ -27,3 +29,20 @@ def test_tiled_bounds_same(epsg):
     with pytest.raises(ValidationError):
         tiled_bounds = TiledBounds(epsg=epsg,
                                    bounds=[single_bound, single_bound])
+
+
+# def test_create_tiled_image_data():
+#     bounds_points = [Point(x=0, y=0), Point(x=5, y=5)]
+#     url = "https://labelbox.s3-us-west-2.amazonaws.com/pathology/{z}/{x}/{y}.png"
+#     zoom_levels = [1, 10]
+
+#     tile_layer = TileLayer(url=url, name="slippy map tile")
+#     tile_bounds = TiledBounds(epsg=EPSG.EPSG4326, bounds=bounds_points)
+#     tiled_image_data = TiledImageData(tile_layer=tile_layer,
+#                                       tile_bounds=tile_bounds,
+#                                       zoom_levels=zoom_levels,
+#                                       version=2)
+#     assert isinstance(tiled_image_data, TiledImageData)
+#     assert tiled_image_data.tile_bounds == bounds_points
+#     assert tiled_image_data.tile_layer.url == url
+#     assert tiled_image_data.zoom_levels == zoom_levels