def test_API(self):
fo = open(TESTFN, 'wb')
fo.write(''' "A", "B", "C"
1, 2, 3.2
7, 4, 1.87''')
fo.close()
s = Sniff(TESTFN)
self.assertEqual(s.comments(), '#')
self.assertEqual(s.delimiter(), ',')
self.assertEqual(s.skiprows(), 1)
self.assertEqual(s.dtype(), {
'names': ('A', 'B', 'C'),
'formats': (float, float, float)
})
x = s.loadtxt()
y = array([(1.0, 2.0, 3.20), (7.0, 4.0, 1.87)],
dtype=[('A', float), ('B', float), ('C', float)])
self.assertNamedClose(x, y)
y = loadtxt(TESTFN, **s.kwds())
self.assertNamedClose(x, y)
y = loadtxt_unknown(TESTFN)
self.assertNamedClose(x, y)
d = array2dict(y)
self.assertEqual(type(d), type({}))
self.assertAllClose(x['A'], [1, 7])
self.assertAllClose(x['B'], [2, 4])
self.assertAllClose(x['C'], [3.2, 1.87])
def test_API(self):
fo = open(TESTFN, 'wb')
fo.write(''' "A", "B", "C"
1, 2, 3.2
7, 4, 1.87''')
fo.close()
s = Sniff(TESTFN)
self.assertEqual(s.comments(), '#')
self.assertEqual(s.delimiter(), ',')
self.assertEqual(s.skiprows(), 1)
self.assertEqual(s.dtype(), {'names': ('A', 'B', 'C'),
'formats': (float, float, float)})
x = s.loadtxt()
y = array([(1.0, 2.0, 3.20),
(7.0, 4.0, 1.87)],
dtype=[('A', float), ('B', float), ('C', float)])
self.assertNamedClose(x, y)
y = loadtxt(TESTFN, **s.kwds())
self.assertNamedClose(x, y)
y = loadtxt_unknown(TESTFN)
self.assertNamedClose(x, y)
d = array2dict(y)
self.assertEqual(type(d), type({}))
self.assertAllClose(x['A'], [1, 7])
self.assertAllClose(x['B'], [2, 4])
self.assertAllClose(x['C'], [3.2, 1.87])
def check(self, name, skip_if_win=False):
"""
Check if the output array from csv_files/<name>.csv
(which is of unkown format)
is the same as the array in csv_files/<name>.py
"""
if skip_if_win and sys.platform.startswith('win'):
raise nose.SkipTest
# Note: The files needed for the test are currently accessed directly.
# This assumes that the files are present, and not in a zipped
# egg. enthought.util.resource as supposed to always work, but
# when it ran the striped the tests in the EPD installer tests,
# it was broken. Since we define zip_safe = False in setup.py
# it is safe to assume the files are always present.
s = Sniff(
os.path.join(os.path.dirname(__file__), 'csv_files',
name + '.csv'))
f_py = os.path.join(os.path.dirname(__file__), 'csv_files',
name + '.py')
if not sys.platform.startswith('win'):
nan = float('nan') # must be in namespace for some .py files
d = eval(open(f_py).read())
self.assertEqual(d['kwds'], s.kwds())
self.assertNamedClose(d['array'], s.loadtxt())
def check(self, name, skip_if_win=False):
"""
Check if the output array from csv_files/<name>.csv
(which is of unkown format)
is the same as the array in csv_files/<name>.py
"""
if skip_if_win and sys.platform.startswith('win'):
raise nose.SkipTest
# Note: The files needed for the test are currently accessed directly.
# This assumes that the files are present, and not in a zipped
# egg. enthought.util.resource as supposed to always work, but
# when it ran the striped the tests in the EPD installer tests,
# it was broken. Since we define zip_safe = False in setup.py
# it is safe to assume the files are always present.
s = Sniff(os.path.join(os.path.dirname(__file__),
'csv_files', name + '.csv'))
f_py = os.path.join(os.path.dirname(__file__),
'csv_files', name + '.py')
if not sys.platform.startswith('win'):
nan = float('nan') # must be in namespace for some .py files
d = eval(open(f_py).read())
self.assertEqual(d['kwds'], s.kwds())
self.assertNamedClose(d['array'], s.loadtxt())
