def run(self): if len(self.args) > 0: self.parser.error("Invalid argument") sure = raw_input("Are you sure? [y/n] ") == "y" if sure: print "Dropping database..." migrations.drop()
def tearDown(self): db.session.remove() migrations.drop() os.remove(self.dbfile) TestCase.tearDown(self)