def database(): # do something before your test _, file_name = tempfile.mkstemp() os.environ['DATABASE_NAME'] = file_name Article.create_table(database_name=file_name) yield # tests runs here # do something after your test os.unlink(file_name)
def database(): # Create a temporary db before the test _, file_name = tempfile.mkstemp() os.environ['DATABASE_NAME'] = file_name Article.create_table(database_name=file_name) yield # Delete the temporary db after the test os.unlink(file_name)
def database(): _, file_name = tempfile.mkstemp() os.environ['DATABASE_NAME'] = file_name Article.create_table(database_name=file_name) yield os.unlink(file_name)
#!/usr/bin/env python3 # -*- coding: utf-8 -*- # File : init_db.py # Author : leejhunhong <*****@*****.**> # Date : 06.05.2021 # Last Modified Date: 06.05.2021 # Last Modified By : leejhunhong <*****@*****.**> if __name__ == '__main__': from blog.models import Article Article.create_table()