fix py3 compatibility

mrstlnos/tests/beam.py

@@ -229,10 +229,11 @@ def main():
 
     slv.start()
 
-    from fwk.testing import *
-    tests = CTests()
-    tests.add(CTest('faked measure', 0, 0))
-    tests.run()
+    # [RB] this code is useless
+    #import fwk.testing as tst 
+    #tests = tst.CTests()
+    #tests.add(tst.CTest('faked measure', 0, 0))
+    #tests.run()
 
 if __name__ == "__main__":
     main()

mrstlnos/utilities.py

@@ -142,17 +142,22 @@ def extract(msh, tag_name, DOFperNode, which_pos, which_dof, ensemble_size, name
     i = 1
     while i < len(nodes):
         if nodes[i] == nodes[i-1]:
-            nodes = np.delete(nodes, i)
+            nodes = np.delete(nodes, i) # [RB] converts nodes(int) to nodes(numpyint64)!
         else:
             i = i + 1
     #print nodes
 
+    # [RB] sorting nodes and/or removing duplicates can be performed
+    # in python with a line such as "nodes = list(set(nodes))"
+    # By doing it this way, numpy is avoided (and the introduction
+    # of numpy 64bits integers which should be casted back to int later)
+
     x = np.empty(np.shape(nodes))
     x_sorted = np.empty(np.shape(nodes))
     nodes_sorted = np.empty(np.shape(nodes))
     sol = np.empty([len(nodes), ensemble_size])
     for i in range(0,len(x)):
-        pos = msh.nodes[nodes[i]-1].pos
+        pos = msh.nodes[int(nodes[i])-1].pos
         x[i] = pos.x[which_pos]
 
     #print x