def _testExcel(self):
if DataTable.canReadExcel():
import sys
sys.stderr = sys.stdout
# print 'Testing Excel...'
xlsfile = os.path.join(os.path.dirname(__file__), 'Sample3.xls')
t = DataTable(xlsfile)
self.assertEqual(t[0][0], 1.0, t[0])
def _testExcel(self):
if DataTable.canReadExcel():
import sys
sys.stderr = sys.stdout
# print 'Testing Excel...'
xlsfile = os.path.join(os.path.dirname(__file__), 'Sample3.xls')
t = DataTable(xlsfile)
self.assertEqual(t[0][0], 1.0, t[0])
def _testExcel(self, usePickleCache=True, removePickleCache=True):
self.assertTrue(DataTable.canReadExcel())
xlsFile = os.path.join(os.path.dirname(__file__), 'Sample.xls')
try:
t = DataTable(xlsFile, usePickleCache=usePickleCache)
finally:
pickleFile = xlsFile + '.pickle.cache'
self.assertIs(os.path.exists(pickleFile), usePickleCache)
if usePickleCache and removePickleCache:
os.remove(pickleFile)
self.assertEqual(t[0][0], 1.0, t[0])
import os
import unittest
from io import StringIO
from MiscUtils.DataTable import DataTable, DataTableError, TableColumn
cannotReadExcel = not DataTable.canReadExcel()
class TestTableColumn(unittest.TestCase):
def testWithType(self):
c = TableColumn('foo:int')
self.assertEqual(c.name(), 'foo')
self.assertTrue(c.type() is int)
def testWithoutType(self):
c = TableColumn('bar')
self.assertEqual(c.name(), 'bar')
self.assertTrue(c.type() is None)
def testWrongSpec(self):
self.assertRaises(DataTableError, TableColumn, 'foo:bar')
self.assertRaises(DataTableError, TableColumn, 'foo:bar:baz')
def testValueForRawValue(self):
c = TableColumn('foo:int')
self.assertEqual(c.valueForRawValue(''), 0)
self.assertEqual(c.valueForRawValue('1'), 1)
self.assertEqual(c.valueForRawValue(2), 2)
self.assertEqual(c.valueForRawValue(2.5), 2)
