def test():
from MDSplus import Tree, Int64, Int64Array, Int32Array, tdi
Tree.setTimeContext() # test initPinoDb
self.assertEqual(Tree.getTimeContext(), (None, None, None))
with Tree(self.tree, self.shot, 'NEW') as ptree:
node = ptree.addNode('S')
ptree.write()
node.tree = Tree(self.tree, self.shot)
for i in range(-9, 9, 3):
d = Int64Array(range(3)) * 10 + i * 10
v = Int32Array(range(3)) + i
node.beginSegment(d[0], d[2], d, v)
self.assertEqual(
node.getSegmentList(20, 59).dim_of(0).tolist(), [0, 30])
self.assertEqual(
node.getSegmentList(20, 60).dim_of(0).tolist(), [0, 30, 60])
self.assertEqual(
node.getSegmentList(21, 60).dim_of(0).tolist(), [30, 60])
self.assertEqual(node.record.data().tolist(), list(range(-9, 9)))
node.tree.setTimeContext(Int64(30), Int64(70), Int64(20))
Tree.setTimeContext(1, 2, 3)
self.assertEqual(node.tree.getTimeContext(), (30, 70, 20))
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
self.assertEqual(node.record.data().tolist(),
[3, 5] + [6]) # delta is applied per segment
node.tree.setTimeContext()
self.assertEqual(node.tree.getTimeContext(), (None, None, None))
self.assertEqual(node.record.data().tolist(), list(range(-9, 9)))
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
tdi('treeopen($,$)', self.tree, self.shot)
Tree.setTimeContext(1, 2, 3) # test privacy to Tree
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
tdi('treeopennew($,$)', self.tree, self.shot + 1)
self.assertEqual(Tree.getTimeContext(), (None, None, None))
Tree.setTimeContext(2, 3, 4) # test privacy to Tree
self.assertEqual(Tree.getTimeContext(), (2, 3, 4))
tdi('treeopen($,$)', self.tree, self.shot)
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
tdi('treeclose()')
self.assertEqual(Tree.getTimeContext(), (2, 3, 4))
tdi('treeclose()', self.tree, self.shot)
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
# t.cleanDatafile()
# t.compressDatafile()
fn = t.getFileName()
t2 = t.copy()
# t.close()
# top = t.top()
# ad = t.addDevice('SUB3')
# an = t.addNode('SUB4')
# createPulse(1)
dir = t.dir()
# gc = t.getCurrent('SUB1')
gd = t.getDefault()
gn = t.getNode("SUB1")
gnwi_iterator = t.getNodeWildIter('***')
gnw_array = t.getNodeWild('***')
gtc_tuple = t.getTimeContext()
im_bool = t.isModified()
# sd_tn = t.setDefault("SUB1")
# ----------------------- Static Methods -----------------------
# gvd_str = t.getVersionDate()
# --------------------------- TreeNode Methods ---------------------------
# from MDSplus import TreeNode
tn = t.getNode("SUB1")
c_tn = tn.copy()
opn_str = tn.ORIGINAL_PART_NAME()
pn_str = tn.PART_NAME()
# cn = tn.children_nids()
# mn = tn.member_nids()
# desc = tn.descendants()
# nod_i = tn.number_of_descendants()g
def time_context(self):
Tree.setTimeContext() # test initPinoDb
self.assertEqual(Tree.getTimeContext(), (None, None, None))
with Tree(self.tree, self.shot + 5, 'NEW') as ptree:
node = ptree.addNode('S')
ptree.write()
ptree.normal()
for i in range(-90, 90, 30):
d = Int64Array(range(30)) * 10 + i * 10
v = Float32Array(range(30)) + i
node.makeSegment(d[0], d[29], d, v)
self.assertEqual(
node.getSegmentList(200, 599).dim_of(0).tolist(), [0, 300])
self.assertEqual(
node.getSegmentList(200, 600).dim_of(0).tolist(), [0, 300, 600])
self.assertEqual(
node.getSegmentList(291, 600).dim_of(0).tolist(), [300, 600])
self.assertEqual(node.record.data().tolist(), list(range(-90, 90)))
sig = node.getXYSignal(xmin=0., xmax=3e-8, num=5)
self.assertEqual(sig.data().tolist(), [0, 1, 2, 3])
self.assertEqual(sig.dim_of().data().tolist(), [0, 10, 20, 30])
node.tree.setTimeContext(Int64(300), Int64(700), Int64(150))
Tree.setTimeContext(1, 2, 3)
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
self.assertEqual(node.tree.getTimeContext(), (300, 700, 150))
set_default_resample_mode("m") # minmax
sig = node.record
self.assertEqual(sig.data().tolist(), [30, 44, 45, 59, 60, 74])
self.assertEqual(sig.dim_of().data().tolist(),
[375, 375, 525, 525, 675, 675])
set_default_resample_mode("i") # interpolation
sig = node.record
self.assertEqual(sig.data().tolist(), [30, 45, 60])
self.assertEqual(sig.dim_of().data().tolist(), [300, 450, 600])
node.setExtendedAttribute("ResampleMode", "Average")
sig = node.record
self.assertEqual(sig.data().tolist(), [37, 52, 67])
self.assertEqual(sig.dim_of().data().tolist(), [375, 525, 675])
node.setExtendedAttribute("ResampleMode", "MinMax")
sig = node.record
self.assertEqual(sig.data().tolist(), [30, 44, 45, 59, 60, 74])
self.assertEqual(sig.dim_of().data().tolist(),
[375, 375, 525, 525, 675, 675])
node.setExtendedAttribute("ResampleMode", "INTERP")
sig = node.record
self.assertEqual(sig.data().tolist(), [30, 45, 60])
self.assertEqual(sig.dim_of().data().tolist(), [300, 450, 600])
node.setExtendedAttribute("ResampleMode", "Previous")
sig = node.record
self.assertEqual(sig.data().tolist(), [30, 44, 60])
self.assertEqual(sig.dim_of().data().tolist(), [300, 450, 600])
node.setExtendedAttribute("ResampleMode", "Closest")
sig = node.record
self.assertEqual(sig.data().tolist(), [30, 45, 60])
self.assertEqual(sig.dim_of().data().tolist(), [300, 450, 600])
node.tree.setTimeContext()
self.assertEqual(node.tree.getTimeContext(), (None, None, None))
self.assertEqual(node.record.data().tolist(), list(range(-90, 90)))
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
tdi('treeopen($,$)', self.tree, self.shot + 5)
Tree.setTimeContext(1, 2, 3) # test privacy to Tree
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
tdi('treeopennew($,$)', self.tree, self.shot + 6)
self.assertEqual(Tree.getTimeContext(), (None, None, None))
Tree.setTimeContext(2, 3, 4) # test privacy to Tree
self.assertEqual(Tree.getTimeContext(), (2, 3, 4))
tdi('treeopen($,$)', self.tree, self.shot + 5)
self.assertEqual(Tree.getTimeContext(), (1, 2, 3))
tdi('treeclose()')
self.assertEqual(Tree.getTimeContext(), (2, 3, 4))
tdi('treeclose()')
self.assertEqual(Tree.getTimeContext(), (2, 3, 4))
