class TestHfrRadialDataHandler(PyonTestCase):

    _hfr_data_handler = None
    _var_list = ['LOND', 'LATD', 'VELU', 'VELV', 'VFLG', 'ESPC', 'ETMP', 'MAXV', 'MINV',
                 'ERSC', 'ERTC', 'XDST', 'YDST', 'RNGE', 'BEAR', 'VELO', 'HEAD', 'SPRC']

    def setUp(self):
        self._hfr_data_handler = HfrRadialDataHandler(data_source='test_data/RDLi_SEAB_2011_08_24_1600.ruv')
        pass

    def tearDown(self):
        self._hfr_data_handler = None
        pass

    def test_get_attributes(self):
        attributes = self._hfr_data_handler.get_attributes()
        #make sure attributes match up here?

    def test_get_attribute(self):
        attributes = self._hfr_data_handler.get_attributes()
        for key in attributes.keys():
            attr_value = self._hfr_data_handler.get_attribute(key)

    def test_get_variables(self):
        variables = self._hfr_data_handler.get_variables()
        #make sure variables match up here?

    def test_acquire_data(self):
        variables = self._var_list
        data_iter = self._hfr_data_handler.acquire_data()
        for vn, slice_, rng, data in data_iter:
            self.assertTrue(vn in variables)
            variables.pop(variables.index(vn))
            self.assertTrue(isinstance(slice_, tuple))
            self.assertTrue(isinstance(rng, tuple))
            self.assertTrue(isinstance(data, numpy.ndarray))
    def setUp(self):

        self._datasets = {}

        file = self._create_tst_data_set('base')

        ext_ds = ExternalDataset(name="base", dataset_description=DatasetDescription(), update_description=UpdateDescription(), contact=ContactInformation())
        ext_ds.dataset_description.parameters["dataset_path"] = file
        ext_ds.dataset_description.parameters["temporal_dimension"] = 'time'

        self._datasets['base'] = HfrRadialDataHandler(data_source=file, ext_dataset=ext_ds)

        file = self._create_tst_data_set('mod_gatt')

        ext_ds = ExternalDataset(name="mod_gatt", dataset_description=DatasetDescription(), update_description=UpdateDescription(), contact=ContactInformation())
        ext_ds.dataset_description.parameters["dataset_path"] = file
        ext_ds.dataset_description.parameters["temporal_dimension"] = 'time'
        self._datasets['mod_gatt'] = HfrRadialDataHandler(data_source=file, ext_dataset=ext_ds)

        file = self._create_tst_data_set('new_gatt')

        ext_ds = ExternalDataset(name="new_gatt", dataset_description=DatasetDescription(), update_description=UpdateDescription(), contact=ContactInformation())
        ext_ds.dataset_description.parameters["dataset_path"] = file
        ext_ds.dataset_description.parameters["temporal_dimension"] = 'time'

        self._datasets['new_gatt'] = HfrRadialDataHandler(data_source=file, ext_dataset=ext_ds)

        file = self._create_tst_data_set('mod_var')

        ext_ds = ExternalDataset(name="mod_var", dataset_description=DatasetDescription(), update_description=UpdateDescription(), contact=ContactInformation())
        ext_ds.dataset_description.parameters["dataset_path"] = file
        ext_ds.dataset_description.parameters["temporal_dimension"] = 'time'
        self._datasets['mod_var'] = HfrRadialDataHandler(data_source=file, ext_dataset=ext_ds)

        file = self._create_tst_data_set('new_var')

        ext_ds = ExternalDataset(name="new_var", dataset_description=DatasetDescription(), update_description=UpdateDescription(), contact=ContactInformation())
        ext_ds.dataset_description.parameters["dataset_path"] = file
        ext_ds.dataset_description.parameters["temporal_dimension"] = 'time'
        self._datasets['new_var'] = HfrRadialDataHandler(data_source=file, ext_dataset=ext_ds)

        file = self._create_tst_data_set('mod_varatt')

        ext_ds = ExternalDataset(name="mod_varatt", dataset_description=DatasetDescription(), update_description=UpdateDescription(), contact=ContactInformation())
        ext_ds.dataset_description.parameters["dataset_path"] = file
        ext_ds.dataset_description.parameters["temporal_dimension"] = 'time'
        self._datasets['mod_varatt'] = HfrRadialDataHandler(data_source=file, ext_dataset=ext_ds)

        #file = 'http://marine.rutgers.edu/cool/maracoos/codar/ooi/radials/ASSA/RDLi_ASSA_2012_02_09_1100.ruv    '
        #ext_ds = ExternalDataset(name="remote", dataset_description=DatasetDescription(), update_description=UpdateDescription(), contact=ContactInformation())
        #ext_ds.dataset_description.parameters["dataset_path"] = file
        #ext_ds.dataset_description.parameters["temporal_dimension"] = 'time'
        #self._datasets['remote'] = HfrRadialDataHandler(data_source=file, ext_dataset=ext_ds)

        pass
 def setUp(self):
     self._hfr_data_handler = HfrRadialDataHandler(data_source='test_data/RDLi_SEAB_2011_08_24_1600.ruv')
     pass