Exemplo n.º 1
0
def test_epsgref():

    ep = epsgRef()
    ep.reset()

    import epsgref

    getprj(4326)
    reload(epsgref)
    from epsgref import prj

    assert 4326 in prj

    ep.add(9999, "junk")
    ep._remove_pyc()  # have to do this in python 2, otherwise won't refresh
    reload(epsgref)
    from epsgref import prj

    assert 9999 in prj

    ep.remove(9999)
    ep._remove_pyc()
    reload(epsgref)
    from epsgref import prj

    assert 9999 not in prj

    ep.reset()
    ep._remove_pyc()
    reload(epsgref)
    from epsgref import prj

    assert len(prj) == 0
Exemplo n.º 2
0
def test_epsgref():
    ep = epsgRef()
    ep.reset()

    import epsgref
    getprj(4326)
    reload(epsgref)
    from epsgref import prj
    assert 4326 in prj

    ep.add(9999, 'junk')
    ep._remove_pyc()  # have to do this in python 2, otherwise won't refresh
    reload(epsgref)
    from epsgref import prj
    assert 9999 in prj

    ep.remove(9999)
    ep._remove_pyc()
    reload(epsgref)
    from epsgref import prj
    assert 9999 not in prj

    ep.reset()
    ep._remove_pyc()
    reload(epsgref)
    from epsgref import prj
    assert len(prj) == 0
Exemplo n.º 3
0
def test_polygon_from_ij():
    """test creation of a polygon from an i, j location using get_vertices()."""
    m = flopy.modflow.Modflow('toy_model', model_ws=mpth)
    botm = np.zeros((2, 10, 10))
    botm[0, :, :] = 1.5
    botm[1, 5, 5] = 4  # negative layer thickness!
    botm[1, 6, 6] = 4
    dis = flopy.modflow.ModflowDis(nrow=10, ncol=10,
                                   nlay=2, delr=100, delc=100,
                                   top=3, botm=botm, model=m)

    ncdf = NetCdf('toy.model.nc', m)
    ncdf.write()

    m.export('toy_model_two.nc')
    dis.export('toy_model_dis.nc')

    mg = m.modelgrid
    mg.set_coord_info(xoff=mg._xul_to_xll(600000.0, -45.0),
                      yoff=mg._yul_to_yll(5170000, -45.0),
                      angrot=-45.0, proj4='EPSG:26715')

    recarray = np.array([(0, 5, 5, .1, True, 's0'),
                         (1, 4, 5, .2, False, 's1'),
                         (0, 7, 8, .3, True, 's2')],
                        dtype=[('k', '<i8'), ('i', '<i8'), ('j', '<i8'),
                               ('stuff', '<f4'), ('stuf', '|b1'),
                               ('stf', np.object)]).view(np.recarray)

    # vertices for a model cell
    geoms = [Polygon(m.modelgrid.get_cell_vertices(i, j)) for i, j in
             zip(recarray.i, recarray.j)]

    assert geoms[0].type == 'Polygon'
    assert np.abs(geoms[0].bounds[-1] - 5169292.893203464) < 1e-4
    fpth = os.path.join(mpth, 'test.shp')
    recarray2shp(recarray, geoms, fpth, epsg=26715)
    ep = epsgRef()
    prj = ep.to_dict()
    assert 26715 in prj
    fpth = os.path.join(mpth, 'test.prj')
    fpth2 = os.path.join(mpth, '26715.prj')
    shutil.copy(fpth, fpth2)
    fpth = os.path.join(mpth, 'test.shp')
    recarray2shp(recarray, geoms, fpth, prj=fpth2)

    # test_dtypes
    fpth = os.path.join(mpth, 'test.shp')
    ra = shp2recarray(fpth)
    assert "int" in ra.dtype['k'].name
    assert "float" in ra.dtype['stuff'].name
    assert "bool" in ra.dtype['stuf'].name
    assert "object" in ra.dtype['stf'].name
    assert True
Exemplo n.º 4
0
def test_epsgref():
    ep = epsgRef()
    ep.reset()

    getprj(4326)
    prj = ep.to_dict()
    assert 4326 in prj

    ep.add(9999, 'junk')
    prj = ep.to_dict()
    assert 9999 in prj

    ep.remove(9999)
    prj = ep.to_dict()
    assert 9999 not in prj

    ep.reset()
    prj = ep.to_dict()
    assert len(prj) == 0
Exemplo n.º 5
0
def test_epsgref():
    ep = epsgRef()
    ep.reset()

    getprj(32614)  # WGS 84 / UTM zone 14N
    prj = ep.to_dict()
    assert 32614 in prj

    ep.add(9999, 'junk')
    prj = ep.to_dict()
    assert 9999 in prj

    ep.remove(9999)
    prj = ep.to_dict()
    assert 9999 not in prj

    ep.reset()
    prj = ep.to_dict()
    assert len(prj) == 0
Exemplo n.º 6
0
import os
import sys
import shutil

from flopy.utils.geometry import Polygon, LineString, Point
from flopy.export.shapefile_utils import recarray2shp, shp2recarray
from flopy.utils.modpathfile import PathlineFile, EndpointFile
from flopy.utils import geometry
from flopy.utils.reference import epsgRef
ep = epsgRef()
ep.reset()


def dis_to_shp():
    pass


def upw_to_shp():
    pass


def uzf_to_shp():
    pass


def wells_to_shp():
    pass


def sfr_to_shp():
    pass