@@ -2002,11 +2002,13 @@ def test_stats(self, example: str, caplog) -> None:
2002
2002
"Percentage valid points" ,
2003
2003
]
2004
2004
2005
+ stat_types = (int , float , np .integer , np .floating )
2006
+
2005
2007
# Full stats
2006
2008
stats = raster .get_stats ()
2007
2009
for name in expected_stats :
2008
2010
assert name in stats
2009
- assert stats .get (name ) is not None
2011
+ assert isinstance ( stats .get (name ), stat_types )
2010
2012
2011
2013
# With mask
2012
2014
inlier_mask = raster .get_mask ()
@@ -2018,10 +2020,21 @@ def test_stats(self, example: str, caplog) -> None:
2018
2020
"Percentage valid inlier points" ,
2019
2021
]:
2020
2022
assert name in stats_masked
2021
- assert stats_masked .get (name ) is not None
2023
+ assert isinstance ( stats_masked .get (name ), stat_types )
2022
2024
stats_masked .pop (name )
2023
2025
assert stats_masked == stats
2024
2026
2027
+ # Empty mask
2028
+ empty_mask = np .ones_like (inlier_mask )
2029
+ stats_masked = raster .get_stats (inlier_mask = empty_mask )
2030
+ for name in [
2031
+ "Valid inlier count" ,
2032
+ "Total inlier count" ,
2033
+ "Percentage inlier points" ,
2034
+ "Percentage valid inlier points" ,
2035
+ ]:
2036
+ assert stats_masked .get (name ) == 0
2037
+
2025
2038
# Single stat
2026
2039
for name in expected_stats :
2027
2040
stat = raster .get_stats (stats_name = name )
0 commit comments