added AppendRBNDVI AppendGBNDVI AppendGRNDVI

microsoft · Mar 2, 2022 · b5c7235 · b5c7235
1 parent 446f8c4
commit b5c7235
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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,
@@ -84,6 +87,9 @@ def test_append_triband_index_batch(batch: Dict[str, Tensor]) -> None:
         AppendNDWI,
         AppendSWI,
         AppendGNDVI,
+        AppendGBNDVI,
+        AppendGRNDVI,
+        AppendRBNDVI,
     ],
 )
 def test_append_normalized_difference_indices(