import pymongo


def setUpMongoDB(db_name="testing", col_name="S&P 500"):
    client = pymongo.MongoClient(
        "mongodb+srv://lzcai:<EMAIL>/test?retryWrites=true&w=majority")
    # db = client['testing']
    # collection = db['S&P500']
    db = client[db_name]
    collection = db[col_name]

    return client, db, collection
import pymongo


def setUpMongoDB(db_name="testing", col_name="S&P 500"):
    client = pymongo.MongoClient(
        "mongodb+srv://lzcai:<EMAIL>/test?retryWrites=true&w=majority")
    # db = client['testing']
    # collection = db['S&P500']
    db = client[db_name]
    collection = db[col_name]

    return client, db, collection
# db = client['testing'] # collection = db['S&P500']