def test_data_frame_max(self):
df2 = data_frame_max(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], 2)
self.assertEqual(df2.iloc[0, 1], 'b')
self.assertEqual(df2.iloc[0, 2], False)
self.assertEqual(df2.iloc[0, 3], 0.5)
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)
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)