map01bf@662: #!/usr/bin/env python
map01bf@662: # encoding: utf-8
map01bf@662: """
map01bf@662: InitialisationRelated.py
map01bf@662: 
map01bf@662: designed to mirror the numbering for the C/C++ api's unit tests
map01bf@662: this performs tests 0001, 0002, 0003
map01bf@662: 
map01bf@662: 
map01bf@662: Created by Ben Fields on 2010-01-11.
map01bf@662: """
map01bf@662: 
map01bf@662: import sys
map01bf@662: import os,os.path
map01bf@662: import pyadb
map01bf@717: import numpy as np
map01bf@662: import struct
map01bf@662: import unittest
map01bf@662: 
map01bf@662: 
map01bf@662: class CreateADBTests(unittest.TestCase):
map01bf@662: 	def setUp(self):
map01bf@662: 		self.adb = pyadb.Pyadb("test.adb")
map01bf@662: 	def test_DBcreation(self):
map01bf@662: 		self.assert_(os.path.exists(self.adb.path))
map01bf@662: 		self.assertRaises(TypeError, pyadb.Pyadb)
map01bf@662: 	def test_DBstatus(self):
map01bf@662: 		try:
map01bf@662: 			self.adb.status()
map01bf@662: 		except:
map01bf@662: 			self.assert_(False)
map01bf@717: 	def test_1DinsertionFromFileSelfQuery(self):
map01bf@662: 		tH = open("testfeature", 'w')
map01bf@662: 		tH.write(struct.pack("=id",1,1))
map01bf@662: 		tH.close()
map01bf@718: 		self.adb.insert("testfeature", key='testfeature')
map01bf@662: 		self.adb.configQuery["seqLength"] = 1
map01bf@662: 		result = self.adb.query("testfeature")
map01bf@662: 		self.assert_(len(result.rawData) == 1)
map01bf@662: 		self.assert_(result.rawData.has_key("testfeature"))
map01bf@662: 		self.assert_(len(result.rawData["testfeature"]) == 1)
mas01mc@760: 		self.assert_(result.rawData["testfeature"][0] == (1.0, 0,0))
map01bf@717: 		os.remove(self.adb.path)#delete the db
map01bf@717: 	def test_1DinsertionFromArraySelfQuery(self):
map01bf@718: 		test1 = np.ones(1)
map01bf@717: 		print "test1: " + str(test1)
map01bf@717: 		self.adb.insert(featData=test1, key="testfeature")
map01bf@717: 		self.adb.configQuery["seqLength"] = 1
map01bf@717: 		result = self.adb.query(key="testfeature")
map01bf@717: 		self.assert_(len(result.rawData) == 1)
map01bf@717: 		self.assert_(result.rawData.has_key("testfeature"))
map01bf@717: 		self.assert_(len(result.rawData["testfeature"]) == 1)
mas01mc@760: 		self.assert_(result.rawData["testfeature"][0] == (1.0, 0,0))
map01bf@662: 		
map01bf@662: 
map01bf@662: 
map01bf@662: if __name__ == '__main__':
mas01mc@760: 	unittest.main()