From 39e0d7d1b98767fe66ccb422dca1931e2e9a237c Mon Sep 17 00:00:00 2001
From: Kel Markert <kmarkert@google.com>
Date: Wed, 24 Jul 2024 13:56:05 -0700
Subject: [PATCH] Fix references to deprecated Landsat Collection 1.

This fixes broken tests in xee.ext_integration_test.py

The tests were broken because the Landsat collection LANDSAT/LC08/C01/T1 was removed from Earth Engine. PR #157 was supposed to fix this but missed some references to "C01". This replaces all references to "C01" with "C02" which is available in the data catalog.

PiperOrigin-RevId: 655690325
---
 xee/ext_integration_test.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/xee/ext_integration_test.py b/xee/ext_integration_test.py
index 27ebc05..ac70af7 100644
--- a/xee/ext_integration_test.py
+++ b/xee/ext_integration_test.py
@@ -54,8 +54,8 @@ def init_ee_for_tests():
   init_params = {
       'opt_url': ee.data.HIGH_VOLUME_API_BASE_URL,
   }
-  if not os.environ.get(_USE_ADC_CREDENTIALS_KEY, False):
-    init_params['credentials'] = _read_identity_pool_creds()
+  # if not os.environ.get(_USE_ADC_CREDENTIALS_KEY, False):
+  #   init_params['credentials'] = _read_identity_pool_creds()
   ee.Initialize(**init_params)
 
 
@@ -319,7 +319,7 @@ def setUp(self):
   def test_guess_can_open__collection_name(self):
     self.assertTrue(self.entry.guess_can_open('LANDSAT/LC08/C02/T1'))
     self.assertFalse(
-        self.entry.guess_can_open('LANDSAT/SomeRandomCollection/C01/T1')
+        self.entry.guess_can_open('LANDSAT/SomeRandomCollection/C02/T1')
     )
     self.assertTrue(self.entry.guess_can_open('ee://LANDSAT/LC08/C02/T1'))
     self.assertTrue(self.entry.guess_can_open('ee:LANDSAT/LC08/C02/T1'))
@@ -336,7 +336,7 @@ def test_guess_can_open__image_collection(self):
 
   def test_open_dataset__sanity_check(self):
     ds = self.entry.open_dataset(
-        pathlib.Path('LANDSAT') / 'LC08' / 'C01' / 'T1',
+        pathlib.Path('LANDSAT') / 'LC08' / 'C02' / 'T1',
         drop_variables=tuple(f'B{i}' for i in range(3, 12)),
         n_images=3,
         projection=ee.Projection('EPSG:4326', [25, 0, 0, 0, -25, 0]),
@@ -345,7 +345,8 @@ def test_open_dataset__sanity_check(self):
     self.assertNotEmpty(dict(ds.coords))
     self.assertEqual(
         list(ds.data_vars.keys()),
-        [f'B{i}' for i in range(1, 3)] + ['BQA'],
+        [f'B{i}' for i in range(1, 3)] +
+        ['QA_PIXEL', 'QA_RADSAT', 'SAA', 'SZA', 'VAA', 'VZA'],
     )
     for v in ds.values():
       self.assertIsNotNone(v.data)
@@ -354,7 +355,7 @@ def test_open_dataset__sanity_check(self):
 
   def test_open_dataset__sanity_check_with_negative_scale(self):
     ds = self.entry.open_dataset(
-        pathlib.Path('LANDSAT') / 'LC08' / 'C01' / 'T1',
+        pathlib.Path('LANDSAT') / 'LC08' / 'C02' / 'T1',
         drop_variables=tuple(f'B{i}' for i in range(3, 12)),
         scale=-25.0,  # in degrees
         n_images=3,
@@ -363,7 +364,8 @@ def test_open_dataset__sanity_check_with_negative_scale(self):
     self.assertNotEmpty(dict(ds.coords))
     self.assertEqual(
         list(ds.data_vars.keys()),
-        [f'B{i}' for i in range(1, 3)] + ['BQA'],
+        [f'B{i}' for i in range(1, 3)] +
+        ['QA_PIXEL', 'QA_RADSAT', 'SAA', 'SZA', 'VAA', 'VZA'],
     )
     for v in ds.values():
       self.assertIsNotNone(v.data)