Codebase list netcdf4-python / run/43bb9c03-4637-45fc-8639-eeda6795fb35/main test / tst_open_mem.py
run/43bb9c03-4637-45fc-8639-eeda6795fb35/main

Tree @run/43bb9c03-4637-45fc-8639-eeda6795fb35/main (Download .tar.gz)

tst_open_mem.py @run/43bb9c03-4637-45fc-8639-eeda6795fb35/mainraw · history · blame

import os
import unittest
import netCDF4

CURRENT_DIR = os.path.dirname(os.path.realpath(__file__))


class TestOpenMem(unittest.TestCase):
    def test_mem_open(self):
        fpath = os.path.join(CURRENT_DIR, "netcdf_dummy_file.nc")

        with open(fpath, 'rb') as f:
            nc_bytes = f.read()

            if not netCDF4.__has_nc_open_mem__:
                with self.assertRaises(ValueError):
                    netCDF4.Dataset('foo_bar', memory=nc_bytes)
                return

            # Needs: https://github.com/Unidata/netcdf-c/pull/400
            if netCDF4.__netcdf4libversion__ < '4.4.1.2':
                with self.assertRaises(OSError):
                    netCDF4.Dataset('foo_bar', memory=nc_bytes)
                return

            with netCDF4.Dataset('foo_bar', memory=nc_bytes) as nc:
                assert nc.filepath() == 'foo_bar'
                assert nc.project_summary == 'Dummy netCDF file'

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