Esempio n. 1
0
 def test_aliases(self):
     self.assertEqual(cf.log_level(), cf.LOG_LEVEL())
     self.assertEqual(cf.free_memory(), cf.FREE_MEMORY())
     self.assertEqual(cf.free_memory_factor(), cf.FREE_MEMORY_FACTOR())
     self.assertEqual(cf.fm_threshold(), cf.FM_THRESHOLD())
     self.assertEqual(cf.total_memory(), cf.TOTAL_MEMORY())
     self.assertEqual(cf.regrid_logging(), cf.REGRID_LOGGING())
     self.assertEqual(cf.relaxed_identities(), cf.RELAXED_IDENTITIES())
     self.assertEqual(cf.tempdir(), cf.TEMPDIR())
     self.assertEqual(cf.chunksize(), cf.CHUNKSIZE())
     self.assertEqual(cf.set_performance(), cf.SET_PERFORMANCE())
     self.assertEqual(cf.of_fraction(), cf.OF_FRACTION())
     self.assertEqual(cf.collapse_parallel_mode(),
                      cf.COLLAPSE_PARALLEL_MODE())
Esempio n. 2
0
    def test_PP_WGDOS_UNPACKING(self):
        f = cf.read(self.ppfile)[0]

        self.assertEqual(f.data.mean(), 3.8080420658506196)

        array = f.array

        for chunksize in self.chunk_sizes:
            with cf.CHUNKSIZE(chunksize):
                f = cf.read(self.ppfile)[0]

                for fmt in ("NETCDF4", "CFA4"):
                    cf.write(f, tmpfile, fmt=fmt)
                    g = cf.read(tmpfile)[0]

                    self.assertTrue(
                        (f.array == array).all(),
                        "Bad unpacking of PP WGDOS packed data",
                    )

                    self.assertTrue(
                        f.equals(g, verbose=2),
                        "Bad writing/reading. fmt=" + fmt,
                    )