MNT: migrate Python test files to new vtk interface

Author: neutrinoceros

test/HD/FargoPlanet/python/testidefix.py

@@ -9,11 +9,11 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKPolar
+from pytools.vtk_io import readVTK
 import numpy as np
 
-V=readVTKPolar('../data.0001.vtk')
-U=readVTKPolar('data.0001.ref.vtk')
+V=readVTK('../data.0001.vtk', geometry='polar')
+U=readVTK('data.0001.ref.vtk', geometry='polar')
 
 # Compute the error on RHO
 error=np.mean(np.abs(V.data['RHO']-U.data['rho']),axis=(0,1))

test/HD/MachReflection/python/testidefix.py

@@ -9,12 +9,12 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKCart
+from pytools.vtk_io import readVTK
 import numpy as np
 
 
-V=readVTKCart('../data.0001.vtk')
-U=readVTKCart('data.0001.ref.vtk')
+V=readVTK('../data.0001.vtk', geometry='cartesian')
+U=readVTK('data.0001.ref.vtk', geometry='cartesian')
 
 # Compute the error on PRS
 error=np.mean(np.abs(np.log(V.data['RHO'])-np.log(U.data['RHO'])),axis=(0,1))

test/HD/ViscousDisk/python/testidefix.py

@@ -9,7 +9,7 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKSpherical
+from pytools.vtk_io import readVTK
 import numpy as np
 import argparse
 import matplotlib.pyplot as plt
@@ -24,7 +24,7 @@
 args=parser.parse_args()
 
 
-V=readVTKSpherical('../data.0001.vtk')
+V=readVTK('../data.0001.vtk', geometry='spherical')
 
 
 iref=32

test/HD/ViscousFlowPastCylinder/python/testidefix.py

@@ -9,11 +9,11 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKPolar
+from pytools.vtk_io import readVTK
 import numpy as np
 
-V=readVTKPolar('../data.0001.vtk')
-U=readVTKPolar('data.0001.ref.vtk')
+V=readVTK('../data.0001.vtk', geometry='polar')
+U=readVTK('data.0001.ref.vtk', geometry='polar')
 
 # Compute the error on PRS
 error=np.mean(np.abs(V.data['VX2']-U.data['vx2']),axis=(0,1))

test/HD/sod-iso/python/testidefix.py

@@ -9,7 +9,7 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKCart
+from pytools.vtk_io import readVTK
 from pytools import sod
 import argparse
 import numpy as np
@@ -25,7 +25,7 @@
 
 args=parser.parse_args()
 
-V=readVTKCart('../data.0002.vtk')
+V=readVTK('../data.0002.vtk', geometry='cartesian')
 gamma = 1.00000000001
 npts = 5000
 

test/HD/sod/python/testidefix.py

@@ -9,7 +9,7 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKCart
+from pytools.vtk_io import readVTK
 from pytools import sod
 import argparse
 import numpy as np
@@ -25,7 +25,7 @@
 
 args=parser.parse_args()
 
-V=readVTKCart('../data.0002.vtk')
+V=readVTK('../data.0002.vtk', geometry='cartesian')
 gamma = 1.4
 npts = 5000
 

test/MHD/AmbipolarCshock/python/testidefix.py

@@ -11,7 +11,7 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKCart
+from pytools.vtk_io import readVTK
 import argparse
 
 parser = argparse.ArgumentParser()
@@ -41,7 +41,7 @@ def f(x,D):
     return(diff)
 
 # load solution
-V=readVTKCart('../data.0001.vtk')
+V=readVTK('../data.0001.vtk', geometry='cartesian')
 
 DSim=V.data['RHO'].flatten()
 if(V.y.size>V.x.size):

test/MHD/AmbipolarCshock3D/python/testidefix.py

@@ -11,7 +11,7 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKCart
+from pytools.vtk_io import readVTK
 import argparse
 
 parser = argparse.ArgumentParser()
@@ -41,7 +41,7 @@ def f(x,D):
     return(diff)
 
 # load solution
-V=readVTKCart('../data.0001.vtk')
+V=readVTK('../data.0001.vtk', geometry='cartesian')
 
 
 if(V.y.size>V.x.size):

test/MHD/AmbipolarShearingBox/python/testidefix.py

@@ -10,7 +10,7 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKCart
+from pytools.vtk_io import readVTK
 
 rep='../'
 nend=1000
@@ -23,7 +23,7 @@
 
 t=dt*np.arange(0,nend)
 for n in range(nend):
-    V=readVTKCart(rep+'/data.'+'%0*d'%(4,n)+'.vtk')
+    V=readVTK(rep+'/data.'+'%0*d'%(4,n)+'.vtk', geometry='cartesian')
     Bx[n]=np.sqrt(np.mean(np.mean(V.data['BX1']**2,axis=2),axis=0))
 
 plt.figure()

test/MHD/AxisFluxTube/python/testidefix.py

@@ -9,11 +9,11 @@
 import os
 import sys
 sys.path.append(os.getenv("IDEFIX_DIR"))
-from pytools.vtk_io import readVTKSpherical
+from pytools.vtk_io import readVTK
 import numpy as np
 
-V=readVTKSpherical('../data.0001.vtk')
-U=readVTKSpherical('data.0001.ref.vtk')
+V=readVTK('../data.0001.vtk', geometry='spherical')
+U=readVTK('data.0001.ref.vtk', geometry='spherical')
 
 # Compute BRMS
 Brms_ref=np.sqrt(V.data['BX1']**2+V.data['BX2']**2+V.data['BX3']**2)