def test_data_frame_min(self): df2 = data_frame_min(TestDataFrameOps.df, 'D') self.assertIsInstance(df2, pd.DataFrame) self.assertEqual(len(df2), 1) self.assertEqual(list(df2.columns), ['A', 'B', 'C', 'D']) self.assertEqual(df2.iloc[0, 0], 5) self.assertEqual(df2.iloc[0, 1], 'y') self.assertEqual(df2.iloc[0, 2], True) self.assertEqual(df2.iloc[0, 3], 0.1)
def test_fat_ops(self): features = read_test_features() gdf = GeoDataFrame.from_features(features) self.assertIsNotNone(gdf.crs) from cate.ops.data_frame import data_frame_min, data_frame_max df_min = data_frame_min(gdf, 'C') self.assertIsInstance(df_min, gpd.GeoDataFrame) self.assertEqual(len(df_min), 1) self.assertEqual(list(df_min.columns), ['A', 'B', 'C', 'geometry']) self.assertIsInstance(df_min.geometry, gpd.GeoSeries) self.assertIsNotNone(df_min.crs) df_max = data_frame_max(gdf, 'C') self.assertIsInstance(df_max, gpd.GeoDataFrame) self.assertEqual(len(df_max), 1) self.assertEqual(list(df_max.columns), ['A', 'B', 'C', 'geometry']) self.assertIsInstance(df_max.geometry, gpd.GeoSeries) self.assertIsNotNone(df_max.crs)