def testString(self):
v=Field(400)
self.assertEqual(str(v),'uniform 400')
v=Field([400],name="List<scalar>")
self.assertEqual(str(v),'nonuniform List<scalar> 1\n(\n 400\n)\n')
v=Field([400])
self.assertEqual(str(v),'nonuniform 1\n(\n 400\n)\n')
def testLength(self):
v = Field(400)
with pytest.raises(TypeError):
l = len(v)
v = Field([400])
self.assertEqual(len(v), 1)
v = Field(400, length=23)
self.assertEqual(len(v), 23)
def testNumpyConversion(self):
import numpy as np
f=Field(400,length=42)
n=f.toNumpy("nx",[("val",np.float64)])
self.assertEqual(n["val"][23],400.)
f=Field(Vector(400,200,100),length=42)
n=f.toNumpy("nx",[(c,np.float64) for c in "xyz"])
self.assertEqual(n["x"][23],400.)
self.assertEqual(n["y"][13],200.)
self.assertEqual(n.size,42)
def testAccess(self):
v = Field(400)
self.assertEqual(v[23], 400)
v = Field(list(range(0, 101, 10)), name="List<scalar>")
s = sum(v.val)
self.assertEqual(v[2], 20)
v[5] += 1
self.assertEqual(sum(v), s + 1)
v = Field(400, length=23)
self.assertEqual(v[10], 400)
self.assertEqual(v[0], 400)
self.assertEqual(v[22], 400)
with pytest.raises(IndexError):
val = v[-1]
with pytest.raises(IndexError):
val = v[23]
def testNumpyConversion(self):
import numpy as np
f = Field(400, length=42)
n = f.toNumpy("nx", [("val", np.float64)])
self.assertEqual(n["val"][23], 400.)
f = Field(Vector(400, 200, 100), length=42)
n = f.toNumpy("nx", [(c, np.float64) for c in "xyz"])
self.assertEqual(n["x"][23], 400.)
self.assertEqual(n["y"][13], 200.)
self.assertEqual(n.size, 42)
def testAccess(self):
v=Field(list(range(0,101,10)),name="List<scalar>")
s=sum(v.val)
self.assertEqual(v[2],20)
v[5]+=1
self.assertEqual(sum(v),s+1)
def testCompare(self):
self.assertNotEqual(Field(400),Field(300))
self.assertNotEqual(Field(400),Field([400],name="List<scalar>"))
self.assertNotEqual(Field([400],name="List<vector>"),Field([400],name="List<scalar>"))
self.assertEqual(Field(400)==None,False)