def testJoin(self):
"""
Test a join
"""
dbobj = DBObject(driver=self.driver, database=self.database)
query = 'SELECT doubleTable.id, intTable.id, doubleTable.log, intTable.thrice '
query += 'FROM doubleTable, intTable WHERE doubleTable.id = intTable.id'
results = dbobj.get_chunk_iterator(query, chunk_size=10)
dtype = [('id', int), ('id_1', int), ('log', float), ('thrice', int)]
i = 0
for chunk in results:
if i < 90:
self.assertEqual(len(chunk), 10)
for row in chunk:
self.assertEqual(2 * (i + 1), row[0])
self.assertEqual(row[0], row[1])
self.assertAlmostEqual(numpy.log(row[0]), row[2], 6)
self.assertEqual(3 * row[0], row[3])
self.assertEqual(dtype, row.dtype)
i += 1
self.assertEqual(i, 99)
#make sure that we found all the matches whe should have
results = dbobj.execute_arbitrary(query)
self.assertEqual(dtype, results.dtype)
i = 0
for row in results:
self.assertEqual(2 * (i + 1), row[0])
self.assertEqual(row[0], row[1])
self.assertAlmostEqual(numpy.log(row[0]), row[2], 6)
self.assertEqual(3 * row[0], row[3])
i += 1
self.assertEqual(i, 99)
def testSingleTableQuery(self):
"""
Test a query on a single table (using chunk iterator)
"""
dbobj = DBObject(driver=self.driver, database=self.database)
query = 'SELECT id, sqrt FROM doubleTable'
results = dbobj.get_chunk_iterator(query)
dtype = [('id', int), ('sqrt', float)]
i = 1
for chunk in results:
for row in chunk:
self.assertEqual(row[0], i)
self.assertAlmostEqual(row[1], numpy.sqrt(i))
self.assertEqual(dtype, row.dtype)
i += 1
self.assertEqual(i, 201)
def testPassingConnection(self):
"""
Repeat the test from testJoin, but with a DBObject whose connection was passed
directly from another DBObject, to make sure that passing a connection works
"""
dbobj_base = DBObject(driver=self.driver, database=self.database)
dbobj = DBObject(connection=dbobj_base.connection)
query = 'SELECT doubleTable.id, intTable.id, doubleTable.log, intTable.thrice '
query += 'FROM doubleTable, intTable WHERE doubleTable.id = intTable.id'
results = dbobj.get_chunk_iterator(query, chunk_size=10)
dtype = [
('id', int),
('id_1', int),
('log', float),
('thrice', int)]
i = 0
for chunk in results:
if i<90:
self.assertEqual(len(chunk), 10)
for row in chunk:
self.assertEqual(2*(i+1), row[0])
self.assertEqual(row[0], row[1])
self.assertAlmostEqual(numpy.log(row[0]), row[2], 6)
self.assertEqual(3*row[0], row[3])
self.assertEqual(dtype, row.dtype)
i += 1
self.assertEqual(i, 99)
#make sure that we found all the matches whe should have
results = dbobj.execute_arbitrary(query)
self.assertEqual(dtype, results.dtype)
i = 0
for row in results:
self.assertEqual(2*(i+1), row[0])
self.assertEqual(row[0], row[1])
self.assertAlmostEqual(numpy.log(row[0]), row[2], 6)
self.assertEqual(3*row[0], row[3])
i += 1
self.assertEqual(i, 99)
def testDtype(self):
"""
Test that passing dtype to a query works
(also test q query on a single table using .execute_arbitrary() directly
"""
dbobj = DBObject(driver=self.driver, database=self.database)
query = 'SELECT id, log FROM doubleTable'
dtype = [('id', int), ('log', float)]
results = dbobj.execute_arbitrary(query, dtype=dtype)
self.assertEqual(results.dtype, dtype)
for xx in results:
self.assertAlmostEqual(numpy.log(xx[0]), xx[1], 6)
self.assertEqual(len(results), 200)
results = dbobj.get_chunk_iterator(query, chunk_size=10, dtype=dtype)
results.next()
for chunk in results:
self.assertEqual(chunk.dtype, dtype)
def testDtype(self):
"""
Test that passing dtype to a query works
(also test q query on a single table using .execute_arbitrary() directly
"""
dbobj = DBObject(driver=self.driver, database=self.database)
query = 'SELECT id, log FROM doubleTable'
dtype = [('id', int), ('log', float)]
results = dbobj.execute_arbitrary(query, dtype = dtype)
self.assertEqual(results.dtype, dtype)
for xx in results:
self.assertAlmostEqual(numpy.log(xx[0]), xx[1], 6)
self.assertEqual(len(results), 200)
results = dbobj.get_chunk_iterator(query, chunk_size=10, dtype=dtype)
results.next()
for chunk in results:
self.assertEqual(chunk.dtype, dtype)
def testSingleTableQuery(self):
"""
Test a query on a single table (using chunk iterator)
"""
dbobj = DBObject(driver=self.driver, database=self.database)
query = 'SELECT id, sqrt FROM doubleTable'
results = dbobj.get_chunk_iterator(query)
dtype = [
('id', int),
('sqrt', float)]
i = 1
for chunk in results:
for row in chunk:
self.assertEqual(row[0], i)
self.assertAlmostEqual(row[1], numpy.sqrt(i))
self.assertEqual(dtype, row.dtype)
i += 1
self.assertEqual(i, 201)
def testJoin(self):
"""
Test a join
"""
dbobj = DBObject(driver=self.driver, database=self.database)
query = 'SELECT doubleTable.id, intTable.id, doubleTable.log, intTable.thrice '
query += 'FROM doubleTable, intTable WHERE doubleTable.id = intTable.id'
results = dbobj.get_chunk_iterator(query, chunk_size=10)
dtype = [
('id', int),
('id_1', int),
('log', float),
('thrice', int)]
i = 0
for chunk in results:
if i<90:
self.assertEqual(len(chunk), 10)
for row in chunk:
self.assertEqual(2*(i+1), row[0])
self.assertEqual(row[0], row[1])
self.assertAlmostEqual(numpy.log(row[0]), row[2], 6)
self.assertEqual(3*row[0], row[3])
self.assertEqual(dtype, row.dtype)
i += 1
self.assertEqual(i, 99)
#make sure that we found all the matches whe should have
results = dbobj.execute_arbitrary(query)
self.assertEqual(dtype, results.dtype)
i = 0
for row in results:
self.assertEqual(2*(i+1), row[0])
self.assertEqual(row[0], row[1])
self.assertAlmostEqual(numpy.log(row[0]), row[2], 6)
self.assertEqual(3*row[0], row[3])
i += 1
self.assertEqual(i, 99)
def testPassingConnection(self):
"""
Repeat the test from testJoin, but with a DBObject whose connection was passed
directly from another DBObject, to make sure that passing a connection works
"""
dbobj_base = DBObject(driver=self.driver, database=self.database)
dbobj = DBObject(connection=dbobj_base.connection)
query = 'SELECT doubleTable.id, intTable.id, doubleTable.log, intTable.thrice '
query += 'FROM doubleTable, intTable WHERE doubleTable.id = intTable.id'
results = dbobj.get_chunk_iterator(query, chunk_size=10)
dtype = [('id', int), ('id_1', int), ('log', float), ('thrice', int)]
i = 0
for chunk in results:
if i < 90:
self.assertEqual(len(chunk), 10)
for row in chunk:
self.assertEqual(2 * (i + 1), row[0])
self.assertEqual(row[0], row[1])
self.assertAlmostEqual(numpy.log(row[0]), row[2], 6)
self.assertEqual(3 * row[0], row[3])
self.assertEqual(dtype, row.dtype)
i += 1
self.assertEqual(i, 99)
#make sure that we found all the matches whe should have
results = dbobj.execute_arbitrary(query)
self.assertEqual(dtype, results.dtype)
i = 0
for row in results:
self.assertEqual(2 * (i + 1), row[0])
self.assertEqual(row[0], row[1])
self.assertAlmostEqual(numpy.log(row[0]), row[2], 6)
self.assertEqual(3 * row[0], row[3])
i += 1
self.assertEqual(i, 99)
