Files
visual_critical_discourse_a…/tests/test_total_annotated.py
T

20 lines
543 B
Python

from pathlib import Path
from dotenv import load_dotenv
import os
from src.database import (
connect,
total_annotated
)
if __name__ == "__main__":
# prepare env vars
env_path = Path(__file__).parent.parent / "local.env"
assert env_path.exists()
load_dotenv(env_path)
os.environ["MONGO_HOST"] = "localhost"
# connect to database
collection, db, client = connect()
# get visual communication
num_docs = total_annotated(collection)
print(f"number of annotated documents in database: {num_docs}")