def test_get_database(self): from shortcut import get_database, connection db = get_database() self.assertEqual(db.connection, connection) db = get_database(connection) self.assertEqual(db.connection, connection)
def get_database(cls): return get_database(connection)