Source code for tests.test_simple_datacube

import unittest
import earthdaily



class TestEarthDataStore(unittest.TestCase):


    def setUp(self):
        self.eds = earthdaily.EarthDataStore()
        self.pivot = earthdaily.datasets.load_pivot()
        self.pivot_corumba = earthdaily.datasets.load_pivot_corumba()




    def test_sentinel1(self):
        collection = "sentinel-1-rtc"
        datacube = self.eds.datacube(
            collection,
            assets=["vh", "vv"],
            intersects=self.pivot,
            datetime="2022-01",
        )
        self.assertEqual(list(datacube.data_vars.keys()), ["vh", "vv"])




    def test_sentinel2(self):
        collection = "sentinel-2-l2a"
        datacube = self.eds.datacube(
            collection,
            assets=["blue", "green", "red"],
            intersects=self.pivot,
            datetime="2023-07-01",
            mask_with=['native']
        )
        self.assertEqual(
            list(datacube.data_vars.keys()), ["blue", "green", "red"]
        )




if __name__ == "__main__":
    unittest.main()