Пример #1
0
def test_layer_dimensions():
    vt = VectorTile()
    layer = vt.add_layer('test')
    assert layer.dimensions == 2
    with pytest.raises(AttributeError):
        layer.dimensions = 3
    assert layer.dimensions == 2
Пример #2
0
def test_create_spline_feature_3d():
    vt = VectorTile()
    layer = vt.add_layer('test', version=3)
    feature = layer.add_spline_feature(has_elevation=True)
    assert isinstance(feature, SplineFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    assert feature.has_elevation
    assert feature.degree == 2

    scaling = layer.add_attribute_scaling(precision=10.0**-8,
                                          min_value=0.0,
                                          max_value=25.0)
    bad_control_points1 = [[8, 10, 1]]
    knot_values = [0.0, 0.0, 0.0, 1.0, 2.0, 2.0, 2.0]
    knots = FloatList(scaling, knot_values)
    with pytest.raises(Exception):
        feature.add_spline(bad_control_points1, knots)
    bad_control_points2 = [[8, 10, 1], [9, 11, 1], [9, 1], [12, 10, 1]]
    with pytest.raises(IndexError):
        feature.add_spline(bad_control_points2, knots)

    control_points = [[8, 10, 1], [9, 11, 1], [11, 9, 1], [12, 10, 1]]
    feature.add_spline(control_points, knots)

    assert feature.get_splines() == [[control_points, knot_values]]
Пример #3
0
def test_layer_features():
    vt = VectorTile()
    layer = vt.add_layer('test')
    assert len(layer.features) == 0
    assert isinstance(layer.features, list)
    with pytest.raises(AttributeError):
        layer.features = [1, 2]
    assert len(layer.features) == 0
Пример #4
0
def test_feature_properties():
    vt = VectorTile()
    layer = vt.add_layer('test')
    feature = layer.add_point_feature()
    assert isinstance(feature, PointFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    prop = feature.properties
    assert isinstance(prop, FeatureProperties)
    assert feature.properties == {}
    assert prop == {}
    prop['fun'] = 'stuff'
    assert 'fun' in prop
    assert prop['fun'] == 'stuff'
    assert feature.properties['fun'] == 'stuff'
    assert feature.properties == {'fun': 'stuff'}
    # Can set by external dictionary
    prop_dict = {'number': 1, 'bool': True, 'string': 'foo', 'float': 4.1}
    feature.properties = prop_dict
    assert feature.properties == prop_dict
    # Key error on not existant property
    with pytest.raises(KeyError):
        foo = feature.properties['doesnotexist']
    # Type errors on invalid key types
    with pytest.raises(TypeError):
        feature.properties[1.234] = True
    with pytest.raises(TypeError):
        feature.properties[1] = True
    with pytest.raises(TypeError):
        foo = feature.properties[1.234]
    with pytest.raises(TypeError):
        foo = feature.properties[1]
    # During setting invalid properties with bad keys or value types will just be dropped
    prop_dict = {
        'foo': [1, 2, 3],
        'fee': [{
            'a': 'b'
        }, {
            'a': ['c', 'd']
        }],
        1.2341: 'stuff',
        1: 'fish',
        'go': False
    }
    feature.properties = prop_dict
    assert feature.properties != prop_dict
    assert feature.properties == {
        'foo': [1, 2, 3],
        'fee': [{
            'a': 'b'
        }, {
            'a': ['c', 'd']
        }],
        'go': False
    }
Пример #5
0
def test_create_polygon_feature_3d():
    vt = VectorTile()
    layer = vt.add_layer('test')
    # Should not be allowed with version 2 layer
    with pytest.raises(Exception):
        feature = layer.add_polygon_feature(has_elevation=True)
    vt = VectorTile()
    layer = vt.add_layer('test', version=3)
    feature = layer.add_polygon_feature(has_elevation=True)
    assert isinstance(feature, PolygonFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    assert feature.has_elevation
    polygon = [[[0, 0, 1], [10, 0, 1], [10, 10, 1], [0, 10, 1], [0, 0, 1]],
               [[3, 3, 1], [3, 5, 1], [5, 5, 1], [3, 3, 1]]]
    feature.add_ring(polygon[0])
    feature.add_ring(polygon[1])
    assert feature.get_rings() == polygon
    assert feature.get_polygons() == [polygon]

    bad_ring1 = [[0, 0, 1], [1, 0, 1]]
    with pytest.raises(Exception):
        feature.add_ring(bad_ring)
    assert feature.get_polygons() == [polygon]

    bad_ring2 = [[0, 0, 1], [1, 0, 1], [0, 0, 1]]
    with pytest.raises(Exception):
        feature.add_ring(bad_ring2)
    assert feature.get_polygons() == [polygon]

    bad_ring3 = [[0, 0, 1], [1, 0, 1], [1, 1, 1], [1, 1], [0, 0, 1]]
    with pytest.raises(IndexError):
        feature.add_ring(bad_ring3)
    assert feature.get_polygons() == [polygon]

    feature.add_ring(polygon[0])
    feature.add_ring(polygon[1])
    assert feature.get_polygons() == [polygon, polygon]

    # clear current geometry
    feature.clear_geometry()
    assert feature.get_rings() == []
    assert feature.get_polygons() == []

    feature.add_ring(polygon[0])
    feature.add_ring(polygon[1])
    assert feature.get_rings() == polygon
    assert feature.get_polygons() == [polygon]

    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    assert feature.get_rings() == polygon
    assert feature.get_polygons() == [polygon]
    feature.add_ring(polygon[0])
    feature.add_ring(polygon[1])
    assert feature.get_polygons() == [polygon, polygon]
Пример #6
0
def test_create_spline_feature():
    vt = VectorTile()
    layer = vt.add_layer('test')
    feature = layer.add_spline_feature()
    assert isinstance(feature, SplineFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    control_points = [[8, 10], [9, 11], [11, 9], [12, 10]]
    knots = [0.0, 0.0, 0.0, 1.0, 2.0, 2.0, 2.0]
    feature.add_control_points(control_points)
    feature.add_knots(knots)

    assert feature.get_control_points() == control_points
    assert feature.get_knots() == knots
Пример #7
0
def test_create_layer():
    vt = VectorTile()
    layer = vt.add_layer('point')
    assert layer.name == 'point'
    assert layer.version == 2
    assert isinstance(layer, Layer)
    layer = vt.add_layer('point_3', 3)
    assert layer.name == 'point_3'
    assert layer.version == 3
    assert isinstance(layer, Layer)
    layer = vt.add_layer('point_4', 4)
    assert layer.name == 'point_4'
    assert layer.version == 4
    assert isinstance(layer, Layer)
Пример #8
0
def test_feature_id():
    vt = VectorTile()
    layer = vt.add_layer('test', version=2)
    feature = layer.add_point_feature()
    assert feature.id == None
    feature.id = 12
    assert feature.id == 12
    # Fails for a version 2 layer
    with pytest.raises(Exception):
        feature.id = "FeatureName"

    layer = vt.add_layer('test2', version=3)
    feature = layer.add_point_feature()
    assert feature.id == None
    feature.id = 12
    assert feature.id == 12
    feature.id = "FeatureName"
    assert feature.id == "FeatureName"

    data = vt.serialize()
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    assert feature.id == 12
    feature = vt.layers[1].features[0]
    assert feature.id == "FeatureName"
Пример #9
0
def test_create_point_feature():
    vt = VectorTile()
    layer = vt.add_layer('test')
    feature = layer.add_point_feature()
    assert isinstance(feature, PointFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    assert not feature.has_elevation
    feature.add_points([10, 11])
    geometry = feature.get_points()
    assert geometry[0] == [10, 11]
    # add points simply adds to end
    feature.add_points([10, 12])
    feature.add_points([10, 13])
    geometry = feature.get_points()
    assert geometry[1] == [10, 12]
    assert geometry[2] == [10, 13]
    # clear current geometry
    feature.clear_geometry()
    assert feature.get_points() == []
    # This is proper way to add multiple points!
    feature.add_points([[10, 11], [10, 12], [10, 13]])
    assert feature.get_points() == [[10, 11], [10, 12], [10, 13]]

    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    feature.add_points([10, 14])
    assert feature.get_points() == [[10, 11], [10, 12], [10, 13], [10, 14]]
Пример #10
0
def test_create_line_feature():
    vt = VectorTile()
    layer = vt.add_layer('test')
    feature = layer.add_line_string_feature()
    assert isinstance(feature, LineStringFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    line_string = [[10, 11], [10, 12], [10, 13], [10, 14]]
    feature.add_line_string(line_string)
    # note that we pull back possible multi line string here
    assert feature.get_line_strings() == [line_string]

    bad_line_string = [[1, 1]]
    with pytest.raises(Exception):
        feature.add_line_string(bad_line_string)
    assert feature.get_line_strings() == [line_string]

    line_string2 = [[9, 9], [30, 5]]
    feature.add_line_string(line_string2)
    assert feature.get_line_strings() == [line_string, line_string2]

    # clear current geometry
    feature.clear_geometry()
    assert feature.get_line_strings() == []
    feature.add_line_string(line_string)
    assert feature.get_line_strings() == [line_string]

    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    feature.add_line_string(line_string2)
    assert feature.get_line_strings() == [line_string, line_string2]
Пример #11
0
def test_create_line_feature_3d():
    vt = VectorTile()
    layer = vt.add_layer('test')
    # Should raise because is version 2 tile
    with pytest.raises(Exception):
        feature = layer.add_line_string_feature(has_elevation=True)
    vt = VectorTile()
    layer = vt.add_layer('test', version=3)
    feature = layer.add_line_string_feature(has_elevation=True)
    assert isinstance(feature, LineStringFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    assert feature.has_elevation
    line_string = [[10, 11, 12], [10, 12, 13], [10, 13, 14], [10, 14, 15]]
    feature.add_line_string(line_string)
    # note that we pull back possible multi line string here
    assert feature.get_line_strings() == [line_string]

    bad_line_string = [[1, 1, 1]]
    with pytest.raises(Exception):
        feature.add_line_string(bad_line_string)
    assert feature.get_line_strings() == [line_string]
    bad_line_string2 = [[1, 1], [2, 2]]
    with pytest.raises(IndexError):
        feature.add_line_string(bad_line_string2)
    assert feature.get_line_strings() == [line_string]

    line_string2 = [[9, 9, 9], [30, 5, 9]]
    feature.add_line_string(line_string2)
    assert feature.get_line_strings() == [line_string, line_string2]

    # clear current geometry
    feature.clear_geometry()
    assert feature.get_line_strings() == []
    feature.add_line_string(line_string)
    assert feature.get_line_strings() == [line_string]

    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    feature.add_line_string(line_string2)
    assert feature.get_line_strings() == [line_string, line_string2]
Пример #12
0
def test_create_spline_feature_3d_with_elevation_scaling():
    vt = VectorTile()
    layer = vt.add_layer('test', version=3)

    scaling = layer.add_attribute_scaling(precision=10.0**-8,
                                          min_value=0.0,
                                          max_value=25.0)
    knot_values = [0.0, 0.0, 0.0, 1.0, 2.0, 2.0, 2.0]
    knots = FloatList(scaling, knot_values)
    control_points = [[8, 10, -11000.0], [9, 11, 9000.0], [11, 9, 85.1],
                      [12, 10, 1500.74]]

    layer.add_elevation_scaling(precision=10.0**-8,
                                min_value=-11000.0,
                                max_value=9000.0)
    with pytest.raises(Exception):
        feature = layer.add_spline_feature(has_elevation=True)
        feature.add_spline(control_points, knots)

    feature = layer.add_spline_feature(has_elevation=True)
    layer.add_elevation_scaling(precision=10.0**-5,
                                min_value=-11000.0,
                                max_value=9000.0)
    feature.add_spline(control_points, knots)

    out = feature.get_splines()
    assert len(out) == 1
    assert len(out[0]) == 2
    out_control_points = out[0][0]
    out_knot_values = out[0][1]
    assert out_knot_values == knot_values
    assert len(control_points) == len(out_control_points)
    for i in range(len(control_points)):
        assert len(control_points[i]) == len(out_control_points[i])
        assert out_control_points[i][0] == control_points[i][0]
        assert out_control_points[i][1] == control_points[i][1]
        assert out_control_points[i][2] == pytest.approx(control_points[i][2])
Пример #13
0
def load_vector_tile(name):
    divided = name.split('_')
    if divided[1] == 'invalid':
        name = '_'.join(divided[2:])
        filename = os.path.join('tests', 'data', 'invalid', name + '.mvt')
    elif divided[1] == 'valid':
        name = '_'.join(divided[2:])
        filename = os.path.join('tests', 'data', 'valid', name + '.mvt')
    else:
        name = '_'.join(divided[1:])
        filename = os.path.join('tests', 'data', name + '.mvt')
    f = open(filename, 'rb')
    test_data = f.read()
    f.close()
    return VectorTile(test_data)
Пример #14
0
def test_create_point_feature_3d():
    vt = VectorTile()
    layer = vt.add_layer('test')
    ## Should fail first because layer is a version 2 layer
    with pytest.raises(Exception):
        feature = layer.add_point_feature(has_elevation=True)
    vt = VectorTile()
    layer = vt.add_layer('test', version=3)
    feature = layer.add_point_feature(has_elevation=True)
    assert isinstance(feature, PointFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    assert feature.has_elevation
    ## Should fail to add 2 point list
    with pytest.raises(IndexError):
        feature.add_points([10, 11])
    feature.add_points([10, 11, 12])
    geometry = feature.get_points()
    assert geometry[0] == [10, 11, 12]
    # add points simply adds to end
    feature.add_points([10, 12, 13])
    feature.add_points([10, 13, 14])
    geometry = feature.get_points()
    assert geometry[1] == [10, 12, 13]
    assert geometry[2] == [10, 13, 14]
    # clear current geometry
    feature.clear_geometry()
    assert feature.get_points() == []
    # This is proper way to add multiple points!
    feature.add_points([[10, 11, 12], [10, 12, 13], [10, 13, 14]])
    assert feature.get_points() == [[10, 11, 12], [10, 12, 13], [10, 13, 14]]

    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    feature.add_points([10, 14, 15])
    assert feature.get_points() == [[10, 11, 12], [10, 12, 13], [10, 13, 14],
                                    [10, 14, 15]]
Пример #15
0
def test_feature_properties():
    vt = VectorTile()
    layer = vt.add_layer('test')
    feature = layer.add_point_feature()
    assert isinstance(feature, PointFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    prop = feature.properties
    assert isinstance(prop, FeatureProperties)
    assert feature.properties == {}
    assert prop == {}
    prop['fun'] = 'stuff'
    assert 'fun' in prop
    assert prop['fun'] == 'stuff'
    assert feature.properties['fun'] == 'stuff'
    assert feature.properties == {'fun':'stuff'}
    # Can set by external dictionary
    prop_dict = { 'number': 1, 'bool': True, 'string': 'foo', 'float': 4.1 }
    feature.properties = prop_dict
    assert feature.properties == prop_dict
    # Key error on not existant property
    with pytest.raises(KeyError):
        foo = feature.properties['doesnotexist']
    # Type errors on invalid key types
    with pytest.raises(TypeError):
        feature.properties[1.234] = True
    with pytest.raises(TypeError):
        feature.properties[1] = True
    with pytest.raises(TypeError):
        foo = feature.properties[1.234]
    with pytest.raises(TypeError):
        foo = feature.properties[1]
    # During setting invalid properties with bad keys or value types will just be dropped
    prop_dict = {'foo': [1,2,3], 'fee': [{'a':'b'}, {'a':['c','d']}], 1.2341: 'stuff', 1: 'fish', 'go': False, 'double': 2.32432, 'float': Float(23432.3222) }
    prop_dict2 = {'foo': [1,2,3], 'fee': [{'a':'b'}, {'a':['c','d']}], 'go': False, 'double': 2.32432, 'float': 23432.3222 }
    feature.properties = prop_dict
    assert feature.properties != prop_dict
    assert feature.properties == prop_dict2
    
    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    assert feature.properties['foo'] == prop_dict2['foo']
    assert feature.properties['fee'] == prop_dict2['fee']
    assert feature.properties['go'] == prop_dict2['go']
    assert feature.properties['double'] == prop_dict2['double']
    # note change is expected due to float encoding!
    assert feature.properties['float'] == 23432.322265625 
Пример #16
0
def test_create_polygon_feature():
    vt = VectorTile()
    layer = vt.add_layer('test')
    feature = layer.add_polygon_feature()
    assert isinstance(feature, PolygonFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    polygon = [[[0, 0], [10, 0], [10, 10], [0, 10], [0, 0]],
               [[3, 3], [3, 5], [5, 5], [3, 3]]]
    feature.add_ring(polygon[0])
    feature.add_ring(polygon[1])
    assert feature.get_rings() == polygon
    assert feature.get_polygons() == [polygon]

    feature.add_ring(polygon[0])
    feature.add_ring(polygon[1])
    assert feature.get_polygons() == [polygon, polygon]

    # clear current geometry
    feature.clear_geometry()
    assert feature.get_rings() == []
    assert feature.get_polygons() == []

    # Add in opposite order
    feature.add_ring(polygon[1])
    feature.add_ring(polygon[0])
    assert feature.get_rings() == [polygon[1], polygon[0]]
    # First ring in wrong winding order so dropped from polygon output
    assert feature.get_polygons() == [[polygon[0]]]

    # clear current geometry
    feature.clear_geometry()
    assert feature.get_rings() == []
    assert feature.get_polygons() == []

    feature.add_ring(polygon[0])
    feature.add_ring(polygon[1])
    assert feature.get_rings() == polygon
    assert feature.get_polygons() == [polygon]

    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    assert feature.get_rings() == polygon
    assert feature.get_polygons() == [polygon]
    feature.add_ring(polygon[0])
    feature.add_ring(polygon[1])
    assert feature.get_polygons() == [polygon, polygon]
Пример #17
0
def test_layer_extent():
    vt = VectorTile()
    layer = vt.add_layer('test')
    assert layer.extent == 4096
    layer.extent = 8000
    assert layer.extent == 8000
Пример #18
0
def test_decode_vector_tile():
    vt = VectorTile(test_data)
    expected_id = 2
    assert len(vt.layers) == 8
    
    # Test first layer
    layer = vt.layers[0]
    assert isinstance(layer, Layer)
    assert layer.name == 'points'
    assert layer.extent == 4096
    assert layer.dimensions == 2
    assert layer.version == 2
    assert len(layer.features) == 4
    # Test layer features
    for feature in layer.features:
        assert isinstance(feature, PointFeature)
        assert feature.type == 'point'
        assert feature.id == expected_id
        geometry = feature.get_points()
        assert geometry == feature.get_geometry()
        assert isinstance(geometry, list)
        assert len(geometry) == 1
        point = geometry[0]
        assert isinstance(point, list)
        assert len(point) == 2
        assert point[0] == 20
        assert point[1] == 20
        props = feature.properties
        assert isinstance(props, FeatureProperties)
        assert len(props) == 1
        if expected_id == 2:
            assert props['some']
            assert props['some'] == 'attr'
        elif expected_id == 3:
            assert props['some']
            assert props['some'] == 'attr'
        elif expected_id == 4:
            assert props['some']
            assert props['some'] == 'otherattr'
        elif expected_id == 5:
            assert props['otherkey']
            assert props['otherkey'] == 'attr'
        expected_id += 1
    
    # Test second layer
    layer = vt.layers[1]
    assert isinstance(layer, Layer)
    assert layer.name == 'lines'
    assert layer.extent == 4096
    assert layer.dimensions == 2
    assert layer.version == 2
    assert len(layer.features) == 1
    # Test layer features
    feature = layer.features[0]
    assert isinstance(feature, LineStringFeature)
    assert feature.type == 'line_string'
    assert feature.id == expected_id
    expected_id += 1
    geometry = feature.get_line_strings()
    assert geometry == feature.get_geometry()
    assert isinstance(geometry, list)
    assert len(geometry) == 2
    geometry == [[[10,10],[10,20],[20,20]],[[11,11],[12,13]]]
    props = feature.properties
    assert isinstance(props, FeatureProperties)
    assert len(props) == 2
    assert props['highway']
    assert props['highway'] == 'primary'
    assert props['maxspeed']
    assert props['maxspeed'] == 50
    
    # Test third layer
    layer = vt.layers[2]
    assert isinstance(layer, Layer)
    assert layer.name == 'polygons'
    assert layer.extent == 4096
    assert layer.dimensions == 2
    assert layer.version == 2
    assert len(layer.features) == 1
    # Test layer features
    feature = layer.features[0]
    assert isinstance(feature, PolygonFeature)
    assert feature.type == 'polygon'
    assert feature.id == expected_id
    expected_id += 1
    geometry = feature.get_rings()
    multi_polygons = feature.get_polygons()
    assert multi_polygons == feature.get_geometry()
    assert isinstance(geometry, list)
    assert isinstance(multi_polygons, list)
    assert len(multi_polygons) == 1
    assert geometry == multi_polygons[0]
    assert len(geometry) == 2
    assert geometry == [[[0,0],[10,0],[10,10],[0,10],[0,0]],[[3,3],[3,5],[5,5],[3,3]]]
    props = feature.properties
    assert isinstance(props, FeatureProperties)
    assert len(props) == 1
    assert props['natural']
    assert props['natural'] == 'wood'

    # Test fourth layer
    layer = vt.layers[3]
    assert isinstance(layer, Layer)
    assert layer.name == 'splines'
    assert layer.extent == 4096
    assert layer.dimensions == 2
    assert layer.version == 2
    assert len(layer.features) == 1
    # Test layer features
    feature = layer.features[0]
    assert isinstance(feature, SplineFeature)
    assert feature.type == 'spline'
    assert feature.id == expected_id
    expected_id += 1
    control_points = feature.get_control_points()
    assert isinstance(control_points, list)
    assert len(control_points) == 4
    assert control_points == [[8,10],[9,11],[11,9],[12,10]]
    knots = feature.get_knots()
    assert [control_points, knots] == feature.get_geometry()
    assert len(knots) == 14
    assert knots == [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.875, 0.0, 2.0, 0.0, 2.0, 0.0, 2.0]
    props = feature.properties
    assert isinstance(props, FeatureProperties)
    assert len(props) == 1
    assert props['natural']
    assert props['natural'] == 'curve'
    
    # Test fifth layer
    expected_id += 1
    layer = vt.layers[4]
    assert isinstance(layer, Layer)
    assert layer.name == 'points_3d'
    assert layer.extent == 4096
    assert layer.dimensions == 3
    assert layer.version == 2
    assert len(layer.features) == 4
    # Test layer features
    point_z = 10
    for feature in layer.features:
        assert isinstance(feature, PointFeature)
        assert feature.type == 'point'
        assert feature.id == expected_id
        geometry = feature.get_points()
        assert isinstance(geometry, list)
        assert len(geometry) == 1
        point = geometry[0]
        assert isinstance(point, list)
        assert len(point) == 3
        assert point[0] == 20
        assert point[1] == 20
        assert point[2] == point_z
        point_z += 10
        props = feature.properties
        assert isinstance(props, FeatureProperties)
        assert len(props) == 1
        if expected_id == 2:
            assert props['some']
            assert props['some'] == 'attr'
        elif expected_id == 3:
            assert props['some']
            assert props['some'] == 'attr'
        elif expected_id == 4:
            assert props['some']
            assert props['some'] == 'otherattr'
        elif expected_id == 5:
            assert props['otherkey']
            assert props['otherkey'] == 'attr'
        expected_id += 1
    
    # Test sixth layer
    layer = vt.layers[5]
    assert isinstance(layer, Layer)
    assert layer.name == 'lines_3d'
    assert layer.extent == 4096
    assert layer.dimensions == 3
    assert layer.version == 2
    assert len(layer.features) == 1
    # Test layer features
    feature = layer.features[0]
    assert isinstance(feature, LineStringFeature)
    assert feature.type == 'line_string'
    assert feature.id == expected_id
    expected_id += 1
    geometry = feature.get_line_strings()
    assert isinstance(geometry, list)
    assert len(geometry) == 2
    geometry == [[[10,10,10],[10,20,20],[20,20,30]],[[11,11,10],[12,13,20]]]
    props = feature.properties
    assert isinstance(props, FeatureProperties)
    assert len(props) == 2
    assert props['highway']
    assert props['highway'] == 'primary'
    assert props['maxspeed']
    assert props['maxspeed'] == 50
    
    # Test seventh layer
    layer = vt.layers[6]
    assert isinstance(layer, Layer)
    assert layer.name == 'polygons_3d'
    assert layer.extent == 4096
    assert layer.dimensions == 3
    assert layer.version == 2
    assert len(layer.features) == 1
    # Test layer features
    feature = layer.features[0]
    assert isinstance(feature, PolygonFeature)
    assert feature.type == 'polygon'
    assert feature.id == expected_id
    expected_id += 1
    geometry = feature.get_rings()
    multi_polygons = feature.get_polygons()
    assert isinstance(geometry, list)
    assert isinstance(multi_polygons, list)
    assert len(multi_polygons) == 2
    assert geometry[0] == multi_polygons[0][0]
    assert geometry[1] == multi_polygons[1][0]
    assert len(geometry) == 2
    assert geometry == [[[0,0,10],[10,0,20],[10,10,30],[0,10,20],[0,0,10]],[[3,3,20],[3,5,40],[5,5,30],[3,3,20]]]
    props = feature.properties
    assert isinstance(props, FeatureProperties)
    assert len(props) == 1
    assert props['natural']
    assert props['natural'] == 'wood'

    # Test eighth layer
    layer = vt.layers[7]
    assert isinstance(layer, Layer)
    assert layer.name == 'splines_3d'
    assert layer.extent == 4096
    assert layer.dimensions == 3
    assert layer.version == 2
    assert len(layer.features) == 1
    # Test layer features
    feature = layer.features[0]
    assert isinstance(feature, SplineFeature)
    assert feature.type == 'spline'
    assert feature.id == expected_id
    expected_id += 1
    control_points = feature.get_control_points()
    assert isinstance(control_points, list)
    assert len(control_points) == 4
    assert control_points == [[8,10,10],[9,11,11],[11,9,12],[12,10,13]]
    knots = feature.get_knots()
    assert len(knots) == 14
    assert knots == [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.875, 0.0, 2.0, 0.0, 2.0, 0.0, 2.0]
    props = feature.properties
    assert isinstance(props, FeatureProperties)
    assert len(props) == 1
    assert props['natural']
    assert props['natural'] == 'curve'
Пример #19
0
def test_create_spline_feature_fail_v2():
    vt = VectorTile()
    layer = vt.add_layer('test')
    with pytest.raises(Exception):
        feature = layer.add_spline_feature()
Пример #20
0
def test_no_layers():
    vt = VectorTile()
    assert len(vt.serialize()) == 0
Пример #21
0
def test_layer_name():
    vt = VectorTile()
    layer = vt.add_layer('test')
    assert layer.name == 'test'
    layer.name = 'foo'
    assert layer.name == 'foo'
Пример #22
0
def test_feature_attributes_version_3():
    vt = VectorTile()
    layer = vt.add_layer('test', version=3)
    assert layer.version == 3
    feature = layer.add_point_feature()
    assert isinstance(feature, PointFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    prop = feature.attributes
    assert isinstance(prop, FeatureAttributes)
    assert feature.attributes == {}
    assert prop == {}
    prop['fun'] = 'stuff'
    assert 'fun' in prop
    assert prop['fun'] == 'stuff'
    assert feature.attributes['fun'] == 'stuff'
    assert feature.attributes == {'fun': 'stuff'}
    # Can set by external dictionary
    prop_dict = {'number': 1, 'bool': True, 'string': 'foo', 'float': 4.1}
    feature.attributes = prop_dict
    assert feature.attributes == prop_dict
    # Key error on not existant property
    with pytest.raises(KeyError):
        foo = feature.attributes['doesnotexist']
    # Type errors on invalid key types
    with pytest.raises(TypeError):
        feature.attributes[1.234] = True
    with pytest.raises(TypeError):
        feature.attributes[1] = True
    with pytest.raises(TypeError):
        foo = feature.attributes[1.234]
    with pytest.raises(TypeError):
        foo = feature.attributes[1]

    dvalues = [1.0, 2.0, None, 2.3, 4.3, None, None, 15.0, 22.5]
    scaling1 = layer.add_attribute_scaling(precision=10.0**-6,
                                           min_value=1.0,
                                           max_value=25.0)
    assert scaling1.index == 0
    assert scaling1.offset == 0
    assert scaling1.base == 1.0
    assert scaling1.multiplier == 9.5367431640625e-07

    # roughly 10**-8 precision
    scaling2 = layer.add_attribute_scaling(offset=10,
                                           base=8.0,
                                           multiplier=7.450580596923828e-09)
    assert scaling2.index == 1
    assert scaling2.offset == 10
    assert scaling2.base == 8.0
    assert scaling2.multiplier == 7.450580596923828e-09

    flist1 = FloatList(scaling1, dvalues)
    flist2 = FloatList(scaling2, dvalues)

    # During setting invalid attributes with bad keys or value types will just be dropped
    prop_dict = {
        'foo': [1, 2, 3],
        'fee': [{
            'a': 'b'
        }, {
            'a': ['c', 'd']
        }],
        1.2341: 'stuff',
        1: 'fish',
        'go': False,
        'double': 2.32432,
        'float': Float(23432.3222),
        'doubleList': flist1,
        'otherDoubleList': flist2
    }
    prop_dict2 = {
        'foo': [1, 2, 3],
        'fee': [{
            'a': 'b'
        }, {
            'a': ['c', 'd']
        }],
        'go': False,
        'double': 2.32432,
        'float': Float(23432.3222),
        'doubleList': flist1,
        'otherDoubleList': flist2
    }
    geometric_dict = {
        'now': [1, 2, 4, 5, 234],
        'stuff': [{
            'x': 12
        }, None, None, None, {
            'y': 13
        }],
        'dlist': flist2,
        'not': None,
        'this': 8,
        'or': True,
        'that': {
            'lost': 'values'
        }
    }
    geometric_dict2 = {
        'now': [1, 2, 4, 5, 234],
        'stuff': [{
            'x': 12
        }, None, None, None, {
            'y': 13
        }],
        'dlist': flist2
    }
    feature.attributes = prop_dict
    assert feature.attributes != prop_dict
    assert feature.attributes == prop_dict2

    feature.geometric_attributes = geometric_dict
    assert feature.geometric_attributes == geometric_dict2

    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    assert feature.attributes['foo'] == prop_dict2['foo']
    assert feature.attributes['fee'] == prop_dict2['fee']
    assert feature.attributes['go'] == prop_dict2['go']
    assert feature.attributes['double'] == prop_dict2['double']
    # note change is expected due to float encoding!
    assert feature.attributes['float'] == 23432.322265625
    # specialized double encoding with scaling should result in approximate equality
    assert isinstance(feature.attributes['doubleList'], list)
    dlist = feature.attributes['doubleList']
    for i in range(len(dlist)):
        if dvalues[i] is None:
            assert dlist[i] is None
        else:
            assert abs(dvalues[i] - dlist[i]) < 10.0**-6
    assert isinstance(feature.attributes['otherDoubleList'], list)
    dlist = feature.attributes['otherDoubleList']
    for i in range(len(dlist)):
        if dvalues[i] is None:
            assert dlist[i] is None
        else:
            assert abs(dvalues[i] - dlist[i]) < 10.0**-8

    assert feature.geometric_attributes['now'] == geometric_dict2['now']
    assert feature.geometric_attributes['stuff'] == geometric_dict2['stuff']
    dlist = feature.geometric_attributes['dlist']
    for i in range(len(dlist)):
        if dvalues[i] is None:
            assert dlist[i] is None
        else:
            assert abs(dvalues[i] - dlist[i]) < 10.0**-8
Пример #23
0
def test_layer_zoom_x_y():
    vt = VectorTile()
    layer = vt.add_layer('test0', version=2)
    assert layer.x == None
    assert layer.y == None
    assert layer.zoom == None
    with pytest.raises(Exception):
        layer.set_tile_location(zoom=3, x=4, y=2)
    layer = vt.add_layer('test1', version=3)
    assert layer.x == None
    assert layer.y == None
    assert layer.zoom == None
    layer.set_tile_location(zoom=3, x=4, y=2)
    assert layer.x == 4
    assert layer.y == 2
    assert layer.zoom == 3
    layer = vt.add_layer('test2', version=3, x=0, y=0, zoom=0)
    assert layer.x == 0
    assert layer.y == 0
    assert layer.zoom == 0
    layer.set_tile_location(zoom=3, x=4, y=2)
    assert layer.x == 4
    assert layer.y == 2
    assert layer.zoom == 3
    with pytest.raises(Exception):
        layer.x = 5
    with pytest.raises(Exception):
        layer.y = 5
    with pytest.raises(Exception):
        layer.zoom = 4
    with pytest.raises(Exception):
        layer = vt.add_layer('test2', version=2, x=0, y=0, zoom=0)
    with pytest.raises(Exception):
        vt.add_layer('test3', version=3, x=-1, y=0, zoom=0)
    with pytest.raises(Exception):
        vt.add_layer('test4', version=3, x=1, y=0, zoom=0)
    with pytest.raises(Exception):
        vt.add_layer('test5', version=3, x=2, y=0, zoom=1)
    with pytest.raises(Exception):
        vt.add_layer('test3', version=3, x=0, y=-1, zoom=0)
    with pytest.raises(Exception):
        vt.add_layer('test4', version=3, x=0, y=1, zoom=0)
    with pytest.raises(Exception):
        vt.add_layer('test5', version=3, x=0, y=2, zoom=1)
    with pytest.raises(Exception):
        vt.add_layer('test3', version=3, x=0, y=0, zoom=-1)
    with pytest.raises(Exception):
        vt.add_layer('test3', version=3, x=0, y=0, zoom=51)
Пример #24
0
def test_feature_attributes_version_3_legacy():
    vt = VectorTile()
    layer = vt.add_layer('test', version=3, legacy_attributes=True)
    assert layer.version == 3
    feature = layer.add_point_feature()
    assert isinstance(feature, PointFeature)
    assert len(layer.features) == 1
    assert feature == layer.features[0]
    prop = feature.attributes
    assert isinstance(prop, FeatureAttributes)
    assert feature.attributes == {}
    assert prop == {}
    prop['fun'] = 'stuff'
    assert 'fun' in prop
    assert prop['fun'] == 'stuff'
    assert feature.attributes['fun'] == 'stuff'
    assert feature.attributes == {'fun': 'stuff'}
    # Can set by external dictionary
    prop_dict = {'number': 1, 'bool': True, 'string': 'foo', 'float': 4.1}
    feature.attributes = prop_dict
    assert feature.attributes == prop_dict
    # Key error on not existant property
    with pytest.raises(KeyError):
        foo = feature.attributes['doesnotexist']
    # Type errors on invalid key types
    with pytest.raises(TypeError):
        feature.attributes[1.234] = True
    with pytest.raises(TypeError):
        feature.attributes[1] = True
    with pytest.raises(TypeError):
        foo = feature.attributes[1.234]
    with pytest.raises(TypeError):
        foo = feature.attributes[1]
    assert len(layer._values) == 5
    # During setting invalid attributes with bad keys or value types will just be dropped
    prop_dict = {
        'foo': [1, 2, 3],
        'fee': [{
            'a': 'b'
        }, {
            'a': ['c', 'd']
        }],
        1.2341: 'stuff',
        1: 'fish',
        'go': False,
        'double': 2.32432,
        'float': Float(23432.3222),
        'double2': 23432.3222,
        'double3': 23432.3222
    }
    prop_dict2 = {
        'go': False,
        'double': 2.32432,
        'float': Float(23432.3222),
        'double2': 23432.3222,
        'double3': 23432.3222
    }
    feature.attributes = prop_dict
    assert feature.attributes != prop_dict
    assert feature.attributes == prop_dict2
    assert len(layer._values) == 9

    # Show that geometric attributes don't work with version 3 legacy attributes
    with pytest.raises(Exception):
        feature.geometric_attributes = {'hmm': [1, 2, 3, 4, 5]}

    # Now serialize the tile
    data = vt.serialize()
    # Reload as new tile to check that cursor moves to proper position for another add point
    vt = VectorTile(data)
    feature = vt.layers[0].features[0]
    assert feature.attributes['go'] == prop_dict2['go']
    assert feature.attributes['double'] == prop_dict2['double']
    # note change is expected due to float encoding!
    assert feature.attributes['float'] == 23432.322265625
    # Show that geometric attributes don't work with version 3 with legacy attributes
    assert feature.geometric_attributes == {}