class TestCsv(unittest.TestCase):
def setUp(self):
self.csv = Csv(['id', 'first name', 'last name'])
self.csv.add([1, 'John', 'Carter'])
self.csv.add([2, 'John', 'Dupond'])
def test_select_equality(self):
records = self.csv.select({'id': 1, 'first name': 'John'}, Csv.EQUALITY)
self.assertTrue(len(records) == 1)
records = self.csv.select({'first name': 'John'}, Csv.EQUALITY)
self.assertTrue(len(records) == 2)
records = self.csv.select({'id': 3, 'first name': 'John'}, Csv.EQUALITY)
self.assertTrue(len(records) == 0)
def test_select_match(self):
records = self.csv.select({'first name': '^J', 'last name': '^C'}, Csv.MATCH)
self.assertTrue(len(records) == 1)
records = self.csv.select({'first name': '^J', 'last name': '^(C|D)'}, Csv.MATCH)
self.assertTrue(len(records) == 2)
records = self.csv.select({'first name': '^T', 'last name': '^(C|D)'}, Csv.MATCH)
self.assertTrue(len(records) == 0)
def test_select_execute(self):
records = self.csv.select({'id': idMatcher}, Csv.EXECUTE)
self.assertTrue(len(records) == 2)
records = self.csv.select({'id': idMatcher, 'first name': firstNameMatcher}, Csv.EXECUTE)
self.assertTrue(len(records) == 2)
def test_len(self):
self.assertTrue(len(self.csv) == 2)
def test_iterator(self):
cpt = 0
for i in self.csv:
cpt += 1
for i in self.csv:
cpt += 1
self.assertTrue(cpt == 2*len(self.csv))
def test_getItem(self):
record = self.csv[1]
id = self.csv.getValue(record, 'id')
self.assertTrue(id == 2)
def test_delItmen(self):
del self.csv[1]
self.assertTrue(len(self.csv) == 1)
record = self.csv[0]
self.assertTrue(record[0] == 1)
def test_setItem(self):
self.csv[0] = [3, 'John', 'Dupond']
record = self.csv[0]
self.assertTrue(record[0] == 3)
def test_keys(self):
v = []
for c in self.csv.keys(): v.append(c)
self.assertTrue(v[0] == 'id')
self.assertTrue(v[1] == 'first name')
self.assertTrue(v[2] == 'last name')
def test_items(self):
v = []
for name, value in self.csv.items(): v.append([name, value])
self.assertTrue(v[0][0] == 'id')
self.assertTrue(v[0][1][0] == 1)
self.assertTrue(v[0][1][1] == 2)
self.assertTrue(v[1][0] == 'first name')
self.assertTrue(v[1][1][0] == 'John')
self.assertTrue(v[1][1][1] == 'John')
def test_values(self):
v = []
for value in self.csv.values(): v.append(value)
self.assertTrue(v[0][0] == 1)
self.assertTrue(v[0][1] == 2)
self.assertTrue(v[1][0] == 'John')
self.assertTrue(v[1][1] == 'John')
def test_raise(self):
self.assertRaises(RuntimeError, Csv, ['id', 'toto', 'toto'])
self.assertRaises(RuntimeError, self.csv.add, [1,2,3,4])
self.assertRaises(RuntimeError, self.csv.select, {'power':3})
self.assertRaises(RuntimeError, self.csv.select, {'id':1}, 4)
def test_str(self):
self.csv.setFormater(formater)
print (str(self.csv))
print (csv)
# Get the second record.
record = csv[1]
print ("This is the first record:")
print (myFormater(record, header))
# Get a record's value by name.
print ("Value 'id' of the second record is %d." % csv.getValue(csv[1], 'id'))
# Get all the columns' names.
print ("CSV's columns:")
for name in csv.keys():
print (" - %s" % name)
# Get all values for each columns.
print ("Values per columns:")
for name, value in csv.items():
print (" - %s: %s" % (name, value))
# Get values (columns's names are omited).
print ("Values per columns (columns's names are omited):")
for value in csv.values():
print (" - %s" % value)
# Print using iterators.
