class IsValidFieldContainer(unittest.TestCase): def setUp(self): self.field = FieldContainer(numpy.random.randn(7, 13), longname="voltage", shortname="U", unit="1V") def testWrongDimension(self): self.field.dimensions[0].data = self.field.dimensions[0].data[:-1] self.assertFalse(self.field.isValid()) def testWrongDimensionNumber(self): self.field.dimensions.append(copy.deepcopy(self.field.dimensions[0])) self.assertFalse(self.field.isValid()) self.field.dimensions = [self.field.dimensions[0]] self.assertFalse(self.field.isValid()) def testWrongMask(self): shape = list(self.field.data.shape) self.field.mask = numpy.ones(shape) self.assertTrue(self.field.isValid()) shape[0] = shape[0] + 1 self.field.mask = numpy.ones(shape) self.assertFalse(self.field.isValid()) def testWrongError(self): shape = list(self.field.data.shape) self.field.error = numpy.zeros(shape) self.assertTrue(self.field.isValid()) shape[0] = shape[0] + 1 self.field.error = numpy.ones(shape) self.assertFalse(self.field.isValid()) def testDimension0HasSameShapeAsField(self): self.field.dimensions[0] = copy.deepcopy(self.field) self.assertFalse(self.field.isValid()) def testDimension1HasSameShapeAsField(self): self.field.dimensions[1] = copy.deepcopy(self.field) self.assertFalse(self.field.isValid())