costdistance_test.py 文件源码

def test_costdistance_infinite(self):
        def zero_one(kv):
            k = kv[0]
            return (k.col == 0 and k.row == 1)

        result = cost_distance(self.raster_rdd,
                               geometries=[Point(13, 13)], max_distance=float('inf'))

        tile = result.to_numpy_rdd().filter(zero_one).first()[1]
        point_distance = tile.cells[0][0][0]
        self.assertTrue(point_distance > 1250000)