def test_extract_data(self): rs = ResultSet(self._get_result_set_file("result_set_file.yml")) granules = self.create_test_granules() self.assertEqual(len(granules), 5) result = rs._extract_granule_data(granules) self.assertEqual(len(result), 5) granules = self.create_test_granules(True) self.assertEqual(len(granules), 2) result = rs._extract_granule_data(granules) self.assertEqual(len(result), 5)