#!/usr/bin/python """ ***GSM package tool. ***Created by A. Mints (2012). Cleans all data from the database. """ import argparse from src.gsmconnectionmanager import GSMConnectionManager from tests.testlib import cleanup_db parser = argparse.ArgumentParser(description=""" ***GSM package tool. ***Created by A. Mints (2012). Cleans all data from the database.""", formatter_class=argparse.RawDescriptionHelpFormatter) parser.add_argument('-D', '--database', type=str, default='test', help='database name to load data into') parser.add_argument('-M', '--monetdb', action="store_true", default=False, help='Use MonetDB instead of PostgreSQL') args = parser.parse_args() cm = GSMConnectionManager(use_monet=args.monetdb, database=args.database) cleanup_db(cm.get_connection()) if args.monetdb: print "MonetDB database %s cleaned" % args.database else: print "PostgreSQL database %s cleaned" % args.database
def setUp(self): super(PipelineGeneralTest, self).setUp() cleanup_db(self.cm.get_connection(database='test')) self.pipeline = GSMPipeline(custom_cm=self.cm, database='test')
def setUp(self): super(SpectraTest, self).setUp() cleanup_db(self.cm.get_connection(database='test')) self.conn = GSMPipeline(custom_cm=self.cm, database='test').conn self.sp = Spectra(self.conn)