diff --git a/tests/transforms/test_indices.py b/tests/transforms/test_indices.py index 1518ca68b42..07584eb672c 100644 --- a/tests/transforms/test_indices.py +++ b/tests/transforms/test_indices.py @@ -10,6 +10,9 @@ from torchgeo.transforms import ( AppendBNDVI, AppendGNDVI, + AppendRBNDVI, + AppendGBNDVI, + AppendGRNDVI, AppendNBR, AppendNDBI, AppendNDRE, @@ -72,6 +75,9 @@ def test_append_triband_index_batch(batch: Dict[str, Tensor]) -> None: AppendNDWI, AppendSWI, AppendGNDVI, + AppendGBNDVI, + AppendGRNDVI, + AppendRBNDVI, ], ) def test_append_normalized_difference_indices(