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
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
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
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
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
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