Fixed a deprecation warning with np.genfromtxt
This commit is contained in:
parent
3df5c60e8f
commit
396d34e1ab
1 changed files with 7 additions and 2 deletions
|
|
@ -278,6 +278,7 @@ def compareCsvFiles(env, file1, file2):
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
hasSkipHeader = tuple(np.version.version.split('.')[:2]) >= ('1','4')
|
||||||
except ImportError:
|
except ImportError:
|
||||||
print 'WARNING: skipping .csv diff because numpy is not installed'
|
print 'WARNING: skipping .csv diff because numpy is not installed'
|
||||||
return 0
|
return 0
|
||||||
|
|
@ -293,8 +294,12 @@ def compareCsvFiles(env, file1, file2):
|
||||||
break
|
break
|
||||||
|
|
||||||
try:
|
try:
|
||||||
data1 = np.genfromtxt(file1, skiprows=headerRows, delimiter=',')
|
if hasSkipHeader:
|
||||||
data2 = np.genfromtxt(file2, skiprows=headerRows, delimiter=',')
|
data1 = np.genfromtxt(file1, skip_header=headerRows, delimiter=',')
|
||||||
|
data2 = np.genfromtxt(file2, skip_header=headerRows, delimiter=',')
|
||||||
|
else:
|
||||||
|
data1 = np.genfromtxt(file1, skiprows=headerRows, delimiter=',')
|
||||||
|
data2 = np.genfromtxt(file2, skiprows=headerRows, delimiter=',')
|
||||||
except (IOError, StopIteration) as e:
|
except (IOError, StopIteration) as e:
|
||||||
print e
|
print e
|
||||||
return 1
|
return 1
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue