class TestPPSReader(unittest.TestCase):
"""Class for testing the PPSReader reader class.
"""
def setUp(self):
self.reader = PPSReader(object)
def test_determine_prod_and_geo_files_local(self):
"""Test the private method _determine_prod_and_geo_files """
self.reader._source = 'local'
self.reader._cloud_product_geodir = None
self.reader._geolocation_product_name = 'CMA'
pro, geo = self.reader._determine_prod_and_geo_files(
CTTH_TESTFILE_LOCAL_1)
self.assertEqual(pro.keys(), ['CTTH'])
self.assertEqual(pro['CTTH'], [CTTH_TESTFILE_LOCAL_1])
self.assertEqual(geo.keys(), ['CTTH'])
self.assertEqual(geo['CTTH'], [GEO_TESTFILE_LOCAL_1])
pro, geo = self.reader._determine_prod_and_geo_files(
[CTTH_TESTFILE_LOCAL_1, CT_TESTFILE_LOCAL_1, CMA_TESTFILE_LOCAL_1])
self.assertEqual(len(pro.keys()), 3)
self.assertTrue('CMA' in pro.keys())
self.assertTrue('CT' in geo.keys())
self.assertTrue('CTTH' in geo.keys())
self.assertEqual(geo['CTTH'], geo['CT'])
self.assertEqual(geo['CTTH'], geo['CMA'])
def test_determine_prod_and_geo_files_ears(self):
"""Test the private method _determine_prod_and_geo_files """
self.reader._source = 'ears'
self.reader._cloud_product_geodir = None
self.reader._geolocation_product_name = None
pro, geo = self.reader._determine_prod_and_geo_files(
CTTH_TESTFILE_EARS_1)
self.assertEqual(pro.keys(), ['CTTH'])
self.assertEqual(pro['CTTH'], [CTTH_TESTFILE_EARS_1])
pro, geo = self.reader._determine_prod_and_geo_files(
[CTTH_TESTFILE_EARS_1, CT_TESTFILE_EARS_1, CMA_TESTFILE_EARS_1])
nkeys = len(pro.keys())
self.assertEqual(nkeys, 3)
for key in pro.keys():
self.assertTrue(key in ["CTTH", "CT", "CMA"])
for key in pro.keys():
self.assertEqual(geo[key], pro[key])
for key in pro.keys():
if key not in ['CMA']:
self.assertNotEqual(geo[key], geo['CMA'])
def tearDown(self):
pass
def setUp(self):
self.reader = PPSReader(object)
