wassname/geopandas
1
0
Fork 0
mirror of https://github.com/wassname/geopandas.git synced 2026-09-17 12:20:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

TEST: Test that checks various numpy int types as columns when writing to file

Browse Source
This commit is contained in:
Kelsey Jordahl
2014-03-25 22:23:24 -04:00
parent 7d932f6c04
commit f443cc7d78
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_geodataframe.py
+11
View File
@@ -268,6 +268,17 @@ class TestDataFrame(unittest.TestCase):
self.assertTrue(len(df) == 5)
self.assertTrue(np.alltrue(df['BoroName'].values == self.boros))
def test_to_file_types(self):
""" Test various integer type columns (GH#93) """
tempfilename = os.path.join(self.tempdir, 'int.shp')
int_types = [np.int, np.int8, np.int16, np.int32, np.int64, np.intp,
np.uint8, np.uint16, np.uint32, np.uint64, np.long]
geometry = self.df2.geometry
data = dict((str(i), np.arange(len(geometry), dtype=dtype))
for i, dtype in enumerate(int_types))
df = GeoDataFrame(data, geometry=geometry)
df.to_file(tempfilename)
def test_mixed_types_to_file(self):
""" Test that mixed geometry types raise error when writing to file """
tempfilename = os.path.join(self.tempdir, 'test.shp')