def test_coord_slice(self): """ Test CoordinateSlicer """ # need some better test cases self.assertTrue(geom_equals(self.g3, self.g3.cx[:, :])) self.assertTrue( geom_equals(self.g3[[True, False]], self.g3.cx[0.9:, :0.1])) self.assertTrue( geom_equals(self.g3[[False, True]], self.g3.cx[0:0.1, 0.9:1.0]))
def _test_binary_topological(self, op, expected, a, b, *args, **kwargs): """ Tests for 'intersection', 'union', 'symmetric_difference', etc. """ if isinstance(expected, GeoPandasBase): fcmp = assert_geoseries_equal else: fcmp = lambda a, b: self.assert_(geom_equals(a, b)) if isinstance(b, GeoPandasBase): right_df = True else: right_df = False self._binary_op_test(op, expected, a, b, fcmp, True, right_df, *args, **kwargs)
def test_coord_slice_with_zero(self): # Test that CoordinateSlice correctly handles zero slice (#GH477). gs = GeoSeries([Point(x, x) for x in range(-3, 4)]) assert geom_equals(gs.cx[:0, :0], gs.loc[:3]) assert geom_equals(gs.cx[:, :0], gs.loc[:3]) assert geom_equals(gs.cx[:0, :], gs.loc[:3]) assert geom_equals(gs.cx[0:, 0:], gs.loc[3:]) assert geom_equals(gs.cx[0:, :], gs.loc[3:]) assert geom_equals(gs.cx[:, 0:], gs.loc[3:])
def _test_binary_operator(self, op, expected, a, b): """ The operators only have GeoSeries on the left, but can have GeoSeries or GeoDataFrame on the right. """ if isinstance(expected, GeoPandasBase): fcmp = assert_geoseries_equal else: fcmp = lambda a, b: self.assert_(geom_equals(a, b)) if isinstance(b, GeoPandasBase): right_df = True else: right_df = False self._binary_op_test(op, expected, a, b, fcmp, False, right_df)
def test_fillna(self): # default is to fill with empty geometry na = self.na_none.fillna() assert isinstance(na[2], BaseGeometry) assert na[2].is_empty assert geom_equals(self.na_none[:2], na[:2])
def fcmp(a, b): assert geom_equals(a, b) if isinstance(b, GeoPandasBase):
def fcmp(a, b): assert geom_equals(a, b)
def test_coord_slice(self): """ Test CoordinateSlicer """ # need some better test cases self.assertTrue(geom_equals(self.g3, self.g3.cx[:, :])) self.assertTrue(geom_equals(self.g3[[True, False]], self.g3.cx[0.9:, :0.1])) self.assertTrue(geom_equals(self.g3[[False, True]], self.g3.cx[0:0.1, 0.9:1.0]))
def test_fillna(self): na = self.na_none.fillna(Point()) self.assertTrue(isinstance(na[2], BaseGeometry)) self.assertTrue(na[2].is_empty) self.assertTrue(geom_equals(self.na_none[:2], na[:2]))