def test_fitness(self):
ds_io = MemDatasetIO()
self.assertEqual(0.75, ds_io.fitness("test.mem", path_type=None))
self.assertEqual(0.75, ds_io.fitness("test.mem", path_type="file"))
self.assertEqual(0.75, ds_io.fitness("test.mem", path_type="dir"))
self.assertEqual(0.75, ds_io.fitness("http://dsio/test.mem", path_type="url"))
self.assertEqual(0.0, ds_io.fitness("test.png", path_type=None))
self.assertEqual(0.0, ds_io.fitness("test.png", path_type="file"))
self.assertEqual(0.0, ds_io.fitness("test.png", path_type="dir"))
self.assertEqual(0.0, ds_io.fitness("http://dsio/test.png", path_type="url"))
ds_io.write(xr.Dataset(), "bibo.odod")
self.assertEqual(1.0, ds_io.fitness("bibo.odod", path_type=None))
self.assertEqual(1.0, ds_io.fitness("bibo.odod", path_type="file"))
self.assertEqual(1.0, ds_io.fitness("bibo.odod", path_type="dir"))
self.assertEqual(1.0, ds_io.fitness("bibo.odod", path_type="url"))
def test_write(self):
ds_io = MemDatasetIO()
ds1 = xr.Dataset()
ds_io.write(ds1, 'test.nc')
ds2 = ds_io._datasets['test.nc']
self.assertIs(ds2, ds1)