Files
visual_critical_discourse_a…/tests/test_get_visual_communication.py
T

21 lines
548 B
Python

from pathlib import Path
from dotenv import load_dotenv
import os
from src.database import (
connect,
get_visual_communication
)
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()
print(client.server_info())
# get visual communication
vis_com = get_visual_communication(collection)
print(vis_com.image)