def test_update(self):
key = "OBS"
obs_size = 4
ens_size = 10
ens_mask = BoolVector(default_value=True, initial_size=ens_size)
block = MeasBlock(key, obs_size, ens_mask)
with self.assertRaises(TypeError):
block["String"] = 10
with self.assertRaises(TypeError):
block[10] = 10
with self.assertRaises(IndexError):
block[obs_size, 0] = 10
with self.assertRaises(IndexError):
block[0, ens_size] = 10
#-----------------------------------------------------------------
with self.assertRaises(TypeError):
a = block["String"]
with self.assertRaises(TypeError):
a = block[10]
with self.assertRaises(IndexError):
val = block[obs_size, 0]
with self.assertRaises(IndexError):
val = block[0, ens_size]
block[1, 2] = 3
self.assertEqual(3, block[1, 2])
def test_inactive(self):
key = "OBS"
obs_size = 2
ens_size = 10
ens_mask = BoolVector(default_value=True, initial_size=ens_size)
ens_mask[5] = False
block = MeasBlock(key, obs_size, ens_mask)
self.assertFalse(block.iensActive(5))
with self.assertRaises(ValueError):
block[0, 5] = 10
def test_create(self):
key = "OBS"
ens_size = 100
obs_size = 77
ens_mask = BoolVector(default_value=True, initial_size=ens_size)
ens_mask[67] = False
block = MeasBlock(key, obs_size, ens_mask)
self.assertEqual(block.getObsSize(), obs_size)
self.assertEqual(block.getActiveEnsSize(), ens_size - 1)
self.assertEqual(block.getTotalEnsSize(), ens_size)
self.assertTrue(block.iensActive(66))
self.assertFalse(block.iensActive(67))
def test_stat(self):
key = "OBS"
obs_size = 2
ens_size = 10
ens_mask = BoolVector(default_value=True, initial_size=ens_size)
block = MeasBlock(key, obs_size, ens_mask)
for iens in range(ens_size):
block[0, iens] = iens
block[1, iens] = iens + 1
self.assertEqual(4.5, block.igetMean(0))
self.assertEqual(5.5, block.igetMean(1))
self.assertFloatEqual(2.872281, block.igetStd(0))
self.assertFloatEqual(2.872281, block.igetStd(1))
