Skip to content

Commit

Permalink
Fix integration tests
Browse files Browse the repository at this point in the history
  • Loading branch information
whimo committed Jun 25, 2024
1 parent d379584 commit bc703e2
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 17 deletions.
1 change: 1 addition & 0 deletions examples/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
# Has to be here to make the examples package importable for integration tests
34 changes: 17 additions & 17 deletions tests/run_integration_tests.py
Original file line number Diff line number Diff line change
@@ -1,35 +1,35 @@
import shutil
from typing import Optional

import os
import argparse
from pathlib import Path
import traceback
import difflib
import json
import os
import shutil
import sys
import traceback
from copy import copy
from functools import partial
from pathlib import Path
from typing import Optional

from dotenv import load_dotenv
import nbformat
from nbconvert.preprocessors import ExecutePreprocessor
from nbformat.v4.nbbase import new_code_cell

from motleycrew.common.exceptions import (
IntegrationTestException,
IpynbIntegrationTestResultNotFound,
)
from motleycrew.common import logger, configure_logging

from dotenv import load_dotenv
from motleycache import (
enable_cache,
set_cache_location,
set_strong_cache,
)
from nbconvert.preprocessors import ExecutePreprocessor
from nbformat.v4.nbbase import new_code_cell

project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
sys.path.insert(0, project_root) # ensure the examples are importable

from examples.old.delegation_crewai import main as delegation_crewai_main
from examples.old.single_llama_index import main as single_llama_index_main

from motleycrew.common import logger, configure_logging
from motleycrew.common.exceptions import (
IntegrationTestException,
IpynbIntegrationTestResultNotFound,
)

INTEGRATION_TESTS = {
"single_llama_index": single_llama_index_main,
Expand Down

0 comments on commit bc703e2

Please sign in to comment.