1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
|
from XdmfCore import *
if __name__ == "__main__":
values1 = [0, 1, 2, 3]
array1 = XdmfArray.New()
array1.insertAsInt8(0, values1)
print array1.getValuesString()
print str(array1.getArrayType()) + " ?= " + str(XdmfArrayType.Int8())
assert array1.getArrayType() == XdmfArrayType.Int8()
array2 = XdmfArray.New()
array2.insertAsInt16(0, values1)
print array2.getValuesString()
print str(array2.getArrayType()) + " ?= " + str(XdmfArrayType.Int16())
assert array2.getArrayType() == XdmfArrayType.Int16()
array3 = XdmfArray.New()
array3.insertAsInt32(0, values1[2:4])
print array3.getValuesString()
print str(array3.getArrayType()) + " ?= " + str(XdmfArrayType.Int32())
assert array3.getArrayType() == XdmfArrayType.Int32()
# Stride
array4 = XdmfArray.New()
array4.insertAsInt64(0, values1[0:4:2])
print array4.getValuesString()
print str(array4.getArrayType()) + " ?= " + str(XdmfArrayType.Int64())
assert array4.getArrayType() == XdmfArrayType.Int64()
# Stride on List
array4a = XdmfArray.New()
array4a.insertAsInt64(0, values1, 2, 4, 2, 1);
print array4a.getValuesString()
print str(array4a.getArrayType()) + " ?= " + str(XdmfArrayType.Int64())
assert array4a.getArrayType() == XdmfArrayType.Int64()
# Reverse
array5 = XdmfArray.New()
array5.insertAsFloat32(0, values1[::-1])
print array5.getValuesString()
print str(array5.getArrayType()) + " ?= " + str(XdmfArrayType.Float32())
assert array5.getArrayType() == XdmfArrayType.Float32()
array6 = XdmfArray.New()
array6.insertAsFloat64(0, values1)
print array6.getValuesString()
print str(array6.getArrayType()) + " ?= " + str(XdmfArrayType.Float64())
assert array6.getArrayType() == XdmfArrayType.Float64()
array7 = XdmfArray.New()
array7.insertAsUInt8(0, values1)
print array7.getValuesString()
print str(array7.getArrayType()) + " ?= " + str(XdmfArrayType.UInt8())
assert array7.getArrayType() == XdmfArrayType.UInt8()
array8 = XdmfArray.New()
array8.insertAsUInt16(0, values1)
print array8.getValuesString()
print str(array8.getArrayType()) + " ?= " + str(XdmfArrayType.UInt16())
assert array8.getArrayType() == XdmfArrayType.UInt16()
array9 = XdmfArray.New()
array9.insertAsUInt32(0, values1)
print array9.getValuesString()
print str(array9.getArrayType()) + " ?= " + str(XdmfArrayType.UInt32())
assert array9.getArrayType() == XdmfArrayType.UInt32()
stringArray = XdmfArray.New()
stringArray.insertValueAsString(0, "foo")
strings = ["bar", "car", "cellar"]
print str(stringArray.getArrayType()) + " ?= " + str(XdmfArrayType.String())
assert stringArray.getArrayType() == XdmfArrayType.String()
stringArray.insertAsString(0, strings)
print str(stringArray.getSize()) + " ?= " + str(3)
print stringArray.getValueAsString(0) + " ?= " + "bar"
assert stringArray.getSize() == 3
assert stringArray.getValueAsString(0) == "bar"
stringArray.pushBackAsString("dog")
print str(stringArray.getSize()) + " ?= " + str(4)
print stringArray.getValueAsString(3) + " ?= " + "dog"
assert stringArray.getSize() == 4
assert stringArray.getValueAsString(3) == "dog"
stringArray.resizeAsString(5, "")
print str(stringArray.getSize()) + " ?= " + str(5)
print stringArray.getValueAsString(4) + " ?= " + ""
assert stringArray.getSize() == 5
assert stringArray.getValueAsString(4) == ""
print stringArray.getValuesString()
# Retrieving data from XdmfArray
sourceArray = XdmfArray.New()
for i in range(0,20):
sourceArray.pushBackAsInt32(i)
output = []
sourceArray.getValues(0, output, sourceArray.getSize())
print output
for i in range(0,20):
assert(output[i] == i)
output2 = []
sourceArray.getValues(0, output2, sourceArray.getSize()/2, 2, 2)
print output2
for i in range(0,len(output2)):
if (i % 2 == 0):
assert(output2[i] == i)
else:
assert(output2[i] == 0)
|