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:
|
||||
import numpy as np
|
||||
hasSkipHeader = tuple(np.version.version.split('.')[:2]) >= ('1','4')
|
||||
except ImportError:
|
||||
print 'WARNING: skipping .csv diff because numpy is not installed'
|
||||
return 0
|
||||
|
|
@ -293,8 +294,12 @@ def compareCsvFiles(env, file1, file2):
|
|||
break
|
||||
|
||||
try:
|
||||
data1 = np.genfromtxt(file1, skiprows=headerRows, delimiter=',')
|
||||
data2 = np.genfromtxt(file2, skiprows=headerRows, delimiter=',')
|
||||
if hasSkipHeader:
|
||||
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:
|
||||
print e
|
||||
return 1
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue