HOME

import os

from xarray import tutorial, DataArray
from xarray.core.pycompat import suppress

from . import TestCase, unittest


@unittest.skip('TODO: make this conditional on network availability')
class Test_load_dataset(TestCase):

    def setUp(self):
        self.testfile = 'tiny'
        self.testfilepath = os.path.expanduser(os.sep.join(
            ('~', '.xarray_tutorial_data', self.testfile)))
        with suppress(OSError):
            os.remove(self.testfilepath)

    def test_download_from_github(self):
        ds = tutorial.load_dataset(self.testfile)
        tiny = DataArray(range(5), name='tiny').to_dataset()
        self.assertDatasetIdentical(ds, tiny)