def testListofStrings(self):
column = ['World', ['Hello', 'World'], 'World']
result = LoadFMF.column2FieldContainer('simple string', column)
expectedResult = FieldContainer(numpy.array(
['World', 'Hello, World', 'World']),
longname='simple string')
assertEqual(result, expectedResult)
def testListofStrings2(self):
column = [['Hello', 'World'], 'World']
result = LoadFMF.column2FieldContainer('simple string', column)
expectedResult = FieldContainer(
numpy.array(['Hello, World', 'World']), longname='simple string'
)
assertEqual(result, expectedResult)
def testStrings(self):
column = ['Hello', 'World']
result = FMFLoader.column2FieldContainer('simple string', column)
expectedResult = FieldContainer(
numpy.array(column), longname='simple string'
)
assertEqual(result, expectedResult)
def testRegionIndexUnit(self):
f = self.field1d.dimensions[0]
mask = slice(3, 7)
afoot = FieldContainer(
self.field1d.data[mask], dimensions=[f[mask]], longname="voltage", shortname="U", unit="1V"
)
self.assertRaises(NotImplementedError, self.field1d.__getitem__, "0.4:0.6")
section = self.field1d["3:7"]
assertEqual(section, afoot)
def testRegionIndexUnit(self):
f = self.field1d.dimensions[0]
mask = slice(3, 7)
afoot = FieldContainer(
self.field1d.data[mask], dimensions=[f[mask]], longname="voltage", shortname="U", unit="1V"
)
section = self.field1d["0.4:0.6A"]
assertEqual(section, afoot)
section = self.field1d["400mA:0.6A"]
assertEqual(section, afoot)
def testVariableFirstNaN(self):
column = [
('T', 'NaN', Quantity('0.5 degC')),
('T', Quantity('11.2 degC'), None)
]
result = LoadFMF.column2FieldContainer('temperature', column)
expectedResult = FieldContainer(
numpy.array([numpy.NaN, 11.2]), error=numpy.array([0.5, 0.0]),
mask = numpy.array([True, False]),
unit='1 degC', longname='temperature', shortname='T')
assertEqual(result, expectedResult)
def testVariable(self):
column = [('T', Quantity('22.4 degC'), Quantity('0.5 degC')),
('T', Quantity('11.2 degC'), Quantity('0.5 degC'))]
result = LoadFMF.column2FieldContainer('temperature', column)
expectedResult = FieldContainer(numpy.array([22.4, 11.2]),
error=numpy.array([0.5, 0.5]),
mask=numpy.array([False, False]),
unit='1 degC',
longname='temperature',
shortname='T')
assertEqual(result, expectedResult)
def testVariableFirstNaN(self):
column = [('T', 'NaN', Quantity('0.5 degC')),
('T', Quantity('11.2 degC'), None)]
result = LoadFMF.column2FieldContainer('temperature', column)
expectedResult = FieldContainer(numpy.array([numpy.NaN, 11.2]),
error=numpy.array([0.5, 0.0]),
mask=numpy.array([True, False]),
unit='1 degC',
longname='temperature',
shortname='T')
assertEqual(result, expectedResult)
def testSingleIndex(self):
section = self.field1d[0]
afoot = FieldContainer(
numpy.array([0.1]),
longname="afoot",
shortname="U",
unit="1V",
dimensions=[self.xDim[0]],
attributes={u"current": (u"I", Quantity("0.3A"))},
)
assertEqual(section, afoot)
def testSingleIndex(self):
section = self.field1d[0]
afoot = FieldContainer(
numpy.array([0.1]),
longname="afoot",
shortname="U",
unit="1V",
dimensions=[],
attributes={u"Index": (u"i", 0)},
)
assertEqual(section, afoot)
def testVariableWithNaN(self):
column = [("T", Quantity("22.4 degC"), Quantity("0.5 degC")), ("T", Quantity("11.2 degC"), None)]
result = LoadFMF.column2FieldContainer("temperature", column)
expectedResult = FieldContainer(
numpy.array([22.4, 11.2]),
error=numpy.array([0.5, 0.0]),
mask=numpy.array([False, False]),
unit="1 degC",
longname="temperature",
shortname="T",
)
assertEqual(result, expectedResult)
def testVariable(self):
column = [
('T', Quantity('22.4 degC'), Quantity('0.5 degC')),
('T', Quantity('11.2 degC'), Quantity('0.5 degC'))
]
result = LoadFMF.column2FieldContainer('temperature', column)
expectedResult = FieldContainer(
numpy.array([22.4, 11.2]) ,error=numpy.array([0.5, 0.5]),
mask=numpy.array([False, False]),
unit='1 degC', longname='temperature', shortname='T'
)
assertEqual(result, expectedResult)
def testRegionIndexUnitRightBoundary(self):
f = self.field1d.dimensions[0]
mask = slice(3, None)
afoot = FieldContainer(
self.field1d.data[mask], dimensions=[f[mask]], longname="voltage", shortname="U", unit="1V"
)
section = self.field1d["0.4:1A"]
assertEqual(section, afoot)
section = self.field1d["0.4A:1000mA"]
assertEqual(section, afoot)
section = self.field1d["0.4A:700mA"]
assertEqual(section, afoot)
def testListofStrings2(self):
column = [["Hello", "World"], "World"]
result = LoadFMF.column2FieldContainer("simple string", column)
expectedResult = FieldContainer(numpy.array(["Hello, World", "World"]), longname="simple string")
assertEqual(result, expectedResult)
