test/test_sqlitecontextmanager.py

import os
from sqlite3 import Cursor

from youtube_transcriber.storing.sqlitecontextmanager import SQLiteContextManager

def test_sqlite_context_manager_init():
    sqlite_context_manager = SQLiteContextManager("dummyinit.db")
    assert type(sqlite_context_manager) == SQLiteContextManager
    
def test_enter_context_manager():
    with SQLiteContextManager("dummy.db") as cursor:
        assert type(cursor) == Cursor
    os.remove("dummy.db")