import sys
import os
from rasdapy import core

db = None

collName = "RasdaPyGrey"

def setUp():
    hostname = "localhost"
    username = "rasadmin"
    password = "rasadmin"
    collName = "RasdaPyGrey"
    con = core.Connection(hostname=hostname,
                            username=username, password=password)
    global db
    db = con.database("RASBASE")
    print db

def test_a():
    global db, collName
    txn = db.transaction(rw=True)
    print "Testing create collection..."
    print collName
    q = txn.query("create collection " + collName + " GreySet")
    res = q.eval()
    txn.commit()
    #assertEqual(res, None)
    print "Done"

def test_b():
    global db, collName
    txn = db.transaction(rw=True)
    print "Testing insert some values to newly created collection..."
    q = txn.query("insert into " + collName + " values marray x in [0:2, 0:2] values 1c")
    res = q.eval()
    txn.commit()
    #assertEqual(res, None)
    print "Done"

def test_c():
    global db, collName
    txn = db.transaction(rw=False)
    print "Testing select some values from newly created collection..."
    q = txn.query("select " + collName + "[0:2,0:2] from " + collName)
    data = q.eval()
    txn.abort()
    #assertIsInstance(data, core.Array)
    # convert rasdaman's output array to numpy array
    # print data.to_array()
    print "Done"

def test_d():
    global db, collName
    txn = db.transaction(rw=False)
    print "Testing get the average cells from newly created collection..."
    q = txn.query("select avg_cells(" + collName + ") from " + collName)
    data = q.eval()
    txn.abort()
    #assertIsInstance(data, list)
    #assertEqual(data, [1.0])
    print "Done"

def test_e():
    global db, collName
    txn = db.transaction(rw=False)
    print "Testing select some values from newly created collection..."
    q = txn.query("select " + collName + "[0:2,0:2] from " + collName)
    data = q.eval()
    txn.abort()
    #assertIsInstance(data, core.Array)
    # convert rasdaman's output array to numpy array
    # print data.to_array()
    print "Done"

def test_f():
    global db, collName
    txn = db.transaction(rw=False)
    print "Testing get the average cells from newly created collection..."
    q = txn.query("select avg_cells(" + collName + ") from " + collName)
    data = q.eval()
    txn.abort()
    #assertIsInstance(data, list)
    #assertEqual(data, [1.0])
    print "Done"

def test_g():
    global db, collName
    txn = db.transaction(rw=False)
    print "Testing select some values from newly created collection..."
    q = txn.query("select " + collName + "[0:2,0:2] from " + collName)
    data = q.eval()
    txn.abort()
    #assertIsInstance(data, core.Array)
    # convert rasdaman's output array to numpy array
    # print data.to_array()
    print "Done"

def test_h():
    global db, collName
    txn = db.transaction(rw=False)
    print "Testing get the average cells from newly created collection..."
    q = txn.query("select avg_cells(" + collName + ") from " + collName)
    data = q.eval()
    txn.abort()
    #assertIsInstance(data, list)
    #assertEqual(data, [1.0])
    print "Done"

def test_i():
    global db, collName
    txn = db.transaction(rw=False)
    print "Testing select some values from newly created collection..."
    q = txn.query("select " + collName + "[0:2,0:2] from " + collName)
    data = q.eval()
    txn.abort()
    #assertIsInstance(data, core.Array)
    # convert rasdaman's output array to numpy array
    # print data.to_array()
    print "Done"

def test_j():
    global db, collName
    txn = db.transaction(rw=False)
    print "Testing get the average cells from newly created collection..."
    q = txn.query("select avg_cells(" + collName + ") from " + collName)
    data = q.eval()
    txn.abort()
    #assertIsInstance(data, list)
    #assertEqual(data, [1.0])
    print "Done"

def test_k():
    global db, collName
    txn = db.transaction(rw=True)
    print "Testing drop collection..."
    q = txn.query("drop collection " + collName)
    res = q.eval()
    txn.commit()
    #assertEqual(res, None)
    print "Done"




setUp()

test_a()
test_b()
test_c()
test_d()
test_e()
test_f()
test_g()
test_h()
test_i()
test_j()
test_k()

# os._exit(1)

print "Trying to close database connection..."
db.close()
print "Done"