From 4989774e85f2e707103b60f310bfad6fe617c868 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Thu, 2 May 2024 16:54:52 -0400 Subject: [PATCH] mark imports --- libs/langchain/tests/unit_tests/callbacks/test_imports.py | 1 - libs/langchain/tests/unit_tests/chains/test_imports.py | 1 - libs/langchain/tests/unit_tests/chat_models/test_imports.py | 1 - libs/langchain/tests/unit_tests/docstore/test_imports.py | 1 - .../tests/unit_tests/document_loaders/test_imports.py | 1 - .../tests/unit_tests/document_transformers/test_imports.py | 1 - libs/langchain/tests/unit_tests/embeddings/test_imports.py | 1 - libs/langchain/tests/unit_tests/load/test_dump.py | 5 +++++ libs/langchain/tests/unit_tests/load/test_serializable.py | 3 +++ .../unit_tests/memory/chat_message_histories/test_imports.py | 1 - libs/langchain/tests/unit_tests/memory/test_imports.py | 1 - .../tests/unit_tests/output_parsers/test_imports.py | 1 - libs/langchain/tests/unit_tests/prompts/test_imports.py | 1 - libs/langchain/tests/unit_tests/retrievers/test_imports.py | 1 - libs/langchain/tests/unit_tests/smith/test_imports.py | 1 - libs/langchain/tests/unit_tests/storage/test_imports.py | 1 - libs/langchain/tests/unit_tests/test_dependencies.py | 1 + libs/langchain/tests/unit_tests/test_imports.py | 3 +++ libs/langchain/tests/unit_tests/tools/test_imports.py | 1 - libs/langchain/tests/unit_tests/utilities/test_imports.py | 1 - libs/langchain/tests/unit_tests/utils/test_imports.py | 1 - 21 files changed, 12 insertions(+), 17 deletions(-) diff --git a/libs/langchain/tests/unit_tests/callbacks/test_imports.py b/libs/langchain/tests/unit_tests/callbacks/test_imports.py index 3e01ae49534..b2ec2bfcb19 100644 --- a/libs/langchain/tests/unit_tests/callbacks/test_imports.py +++ b/libs/langchain/tests/unit_tests/callbacks/test_imports.py @@ -39,4 +39,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(callbacks.__all__) == set(EXPECTED_ALL) - assert_all_importable(callbacks) diff --git a/libs/langchain/tests/unit_tests/chains/test_imports.py b/libs/langchain/tests/unit_tests/chains/test_imports.py index 4da79031fe3..b930fe263bd 100644 --- a/libs/langchain/tests/unit_tests/chains/test_imports.py +++ b/libs/langchain/tests/unit_tests/chains/test_imports.py @@ -68,4 +68,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(chains.__all__) == set(EXPECTED_ALL) - assert_all_importable(chains) diff --git a/libs/langchain/tests/unit_tests/chat_models/test_imports.py b/libs/langchain/tests/unit_tests/chat_models/test_imports.py index e27df46d555..45c9291341a 100644 --- a/libs/langchain/tests/unit_tests/chat_models/test_imports.py +++ b/libs/langchain/tests/unit_tests/chat_models/test_imports.py @@ -37,4 +37,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(chat_models.__all__) == set(EXPECTED_ALL) - assert_all_importable(chat_models) diff --git a/libs/langchain/tests/unit_tests/docstore/test_imports.py b/libs/langchain/tests/unit_tests/docstore/test_imports.py index 763e6d58ff8..f231992409f 100644 --- a/libs/langchain/tests/unit_tests/docstore/test_imports.py +++ b/libs/langchain/tests/unit_tests/docstore/test_imports.py @@ -6,4 +6,3 @@ EXPECTED_ALL = ["DocstoreFn", "InMemoryDocstore", "Wikipedia"] def test_all_imports() -> None: assert set(docstore.__all__) == set(EXPECTED_ALL) - assert_all_importable(docstore) diff --git a/libs/langchain/tests/unit_tests/document_loaders/test_imports.py b/libs/langchain/tests/unit_tests/document_loaders/test_imports.py index 86d5b115da0..4e892aee8a3 100644 --- a/libs/langchain/tests/unit_tests/document_loaders/test_imports.py +++ b/libs/langchain/tests/unit_tests/document_loaders/test_imports.py @@ -177,4 +177,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(document_loaders.__all__) == set(EXPECTED_ALL) - assert_all_importable(document_loaders) diff --git a/libs/langchain/tests/unit_tests/document_transformers/test_imports.py b/libs/langchain/tests/unit_tests/document_transformers/test_imports.py index eadac71fde8..19cc2136fef 100644 --- a/libs/langchain/tests/unit_tests/document_transformers/test_imports.py +++ b/libs/langchain/tests/unit_tests/document_transformers/test_imports.py @@ -19,4 +19,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(document_transformers.__all__) == set(EXPECTED_ALL) - assert_all_importable(document_transformers) diff --git a/libs/langchain/tests/unit_tests/embeddings/test_imports.py b/libs/langchain/tests/unit_tests/embeddings/test_imports.py index 0a5c67c704b..3d93b584cc4 100644 --- a/libs/langchain/tests/unit_tests/embeddings/test_imports.py +++ b/libs/langchain/tests/unit_tests/embeddings/test_imports.py @@ -61,4 +61,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(embeddings.__all__) == set(EXPECTED_ALL) - assert_all_importable(embeddings) diff --git a/libs/langchain/tests/unit_tests/load/test_dump.py b/libs/langchain/tests/unit_tests/load/test_dump.py index 0f2330bc2b9..de314dcd171 100644 --- a/libs/langchain/tests/unit_tests/load/test_dump.py +++ b/libs/langchain/tests/unit_tests/load/test_dump.py @@ -74,6 +74,7 @@ def test_typeerror() -> None: ) +@pytest.mark.community @pytest.mark.requires("openai") def test_serialize_openai_llm(snapshot: Any) -> None: from langchain_community.llms.openai import OpenAI @@ -90,6 +91,7 @@ def test_serialize_openai_llm(snapshot: Any) -> None: assert dumps(llm, pretty=True) == snapshot +@pytest.mark.community @pytest.mark.requires("openai") def test_serialize_llmchain(snapshot: Any) -> None: from langchain_community.llms.openai import OpenAI @@ -100,6 +102,7 @@ def test_serialize_llmchain(snapshot: Any) -> None: assert dumps(chain, pretty=True) == snapshot +@pytest.mark.community @pytest.mark.requires("openai") def test_serialize_llmchain_env() -> None: from langchain_community.llms.openai import OpenAI @@ -124,6 +127,7 @@ def test_serialize_llmchain_env() -> None: del os.environ["OPENAI_API_KEY"] +@pytest.mark.community @pytest.mark.requires("openai") def test_serialize_llmchain_chat(snapshot: Any) -> None: from langchain_community.chat_models.openai import ChatOpenAI @@ -153,6 +157,7 @@ def test_serialize_llmchain_chat(snapshot: Any) -> None: del os.environ["OPENAI_API_KEY"] +@pytest.mark.community @pytest.mark.requires("openai") def test_serialize_llmchain_with_non_serializable_arg(snapshot: Any) -> None: from langchain_community.llms.openai import OpenAI diff --git a/libs/langchain/tests/unit_tests/load/test_serializable.py b/libs/langchain/tests/unit_tests/load/test_serializable.py index ea21e709acc..93e050457b9 100644 --- a/libs/langchain/tests/unit_tests/load/test_serializable.py +++ b/libs/langchain/tests/unit_tests/load/test_serializable.py @@ -1,5 +1,6 @@ import importlib import inspect +import pytest import pkgutil from types import ModuleType @@ -54,6 +55,7 @@ def import_all_modules(package_name: str) -> dict: return classes +@pytest.mark.community def test_import_all_modules() -> None: """Test import all modules works as expected""" all_modules = import_all_modules("langchain") @@ -75,6 +77,7 @@ def test_import_all_modules() -> None: ] +@pytest.mark.community def test_serializable_mapping() -> None: to_skip = { # This should have had a different namespace, as it was never diff --git a/libs/langchain/tests/unit_tests/memory/chat_message_histories/test_imports.py b/libs/langchain/tests/unit_tests/memory/chat_message_histories/test_imports.py index 270014b6f2f..7ed16a6b927 100644 --- a/libs/langchain/tests/unit_tests/memory/chat_message_histories/test_imports.py +++ b/libs/langchain/tests/unit_tests/memory/chat_message_histories/test_imports.py @@ -27,4 +27,3 @@ EXPECTED_ALL = [ def test_imports() -> None: assert sorted(chat_message_histories.__all__) == sorted(EXPECTED_ALL) - assert_all_importable(chat_message_histories) diff --git a/libs/langchain/tests/unit_tests/memory/test_imports.py b/libs/langchain/tests/unit_tests/memory/test_imports.py index e4d819ab890..c4c47cdd9e5 100644 --- a/libs/langchain/tests/unit_tests/memory/test_imports.py +++ b/libs/langchain/tests/unit_tests/memory/test_imports.py @@ -42,4 +42,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(memory.__all__) == set(EXPECTED_ALL) - assert_all_importable(memory) diff --git a/libs/langchain/tests/unit_tests/output_parsers/test_imports.py b/libs/langchain/tests/unit_tests/output_parsers/test_imports.py index a8d1e50f2c0..9cdd8aeb97b 100644 --- a/libs/langchain/tests/unit_tests/output_parsers/test_imports.py +++ b/libs/langchain/tests/unit_tests/output_parsers/test_imports.py @@ -30,4 +30,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(output_parsers.__all__) == set(EXPECTED_ALL) - assert_all_importable(output_parsers) diff --git a/libs/langchain/tests/unit_tests/prompts/test_imports.py b/libs/langchain/tests/unit_tests/prompts/test_imports.py index b66e1a47584..bab33255497 100644 --- a/libs/langchain/tests/unit_tests/prompts/test_imports.py +++ b/libs/langchain/tests/unit_tests/prompts/test_imports.py @@ -27,4 +27,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(prompts.__all__) == set(EXPECTED_ALL) - assert_all_importable(prompts) diff --git a/libs/langchain/tests/unit_tests/retrievers/test_imports.py b/libs/langchain/tests/unit_tests/retrievers/test_imports.py index 935f62bfa24..f7c8b836ca9 100644 --- a/libs/langchain/tests/unit_tests/retrievers/test_imports.py +++ b/libs/langchain/tests/unit_tests/retrievers/test_imports.py @@ -54,4 +54,3 @@ EXPECTED_ALL = [ def test_imports() -> None: assert sorted(retrievers.__all__) == sorted(EXPECTED_ALL) - assert_all_importable(retrievers) diff --git a/libs/langchain/tests/unit_tests/smith/test_imports.py b/libs/langchain/tests/unit_tests/smith/test_imports.py index c8300d8a690..d81541e5be4 100644 --- a/libs/langchain/tests/unit_tests/smith/test_imports.py +++ b/libs/langchain/tests/unit_tests/smith/test_imports.py @@ -10,4 +10,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(smith.__all__) == set(EXPECTED_ALL) - assert_all_importable(smith) diff --git a/libs/langchain/tests/unit_tests/storage/test_imports.py b/libs/langchain/tests/unit_tests/storage/test_imports.py index 33f74105f9d..749a4617dc8 100644 --- a/libs/langchain/tests/unit_tests/storage/test_imports.py +++ b/libs/langchain/tests/unit_tests/storage/test_imports.py @@ -17,4 +17,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(storage.__all__) == set(EXPECTED_ALL) - assert_all_importable(storage) diff --git a/libs/langchain/tests/unit_tests/test_dependencies.py b/libs/langchain/tests/unit_tests/test_dependencies.py index d79879dcf03..32f45562161 100644 --- a/libs/langchain/tests/unit_tests/test_dependencies.py +++ b/libs/langchain/tests/unit_tests/test_dependencies.py @@ -94,6 +94,7 @@ def test_test_group_dependencies(poetry_conf: Mapping[str, Any]) -> None: ) +@pytest.mark.community def test_imports() -> None: """Test that you can import all top level things okay.""" from langchain_community.callbacks import OpenAICallbackHandler # noqa: F401 diff --git a/libs/langchain/tests/unit_tests/test_imports.py b/libs/langchain/tests/unit_tests/test_imports.py index 76ded5305e0..a7dc7980bce 100644 --- a/libs/langchain/tests/unit_tests/test_imports.py +++ b/libs/langchain/tests/unit_tests/test_imports.py @@ -4,7 +4,10 @@ from pathlib import Path # Attempt to recursively import all modules in langchain PKG_ROOT = Path(__file__).parent.parent.parent +import pytest + +@pytest.mark.community def test_import_all() -> None: """Generate the public API for this package.""" library_code = PKG_ROOT / "langchain" diff --git a/libs/langchain/tests/unit_tests/tools/test_imports.py b/libs/langchain/tests/unit_tests/tools/test_imports.py index 79eea82a5ef..163d1d1e3ab 100644 --- a/libs/langchain/tests/unit_tests/tools/test_imports.py +++ b/libs/langchain/tests/unit_tests/tools/test_imports.py @@ -126,4 +126,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(tools.__all__) == set(EXPECTED_ALL) - assert_all_importable(tools) diff --git a/libs/langchain/tests/unit_tests/utilities/test_imports.py b/libs/langchain/tests/unit_tests/utilities/test_imports.py index dcfb2be3178..7d26c4f2dc2 100644 --- a/libs/langchain/tests/unit_tests/utilities/test_imports.py +++ b/libs/langchain/tests/unit_tests/utilities/test_imports.py @@ -53,4 +53,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(utilities.__all__) == set(EXPECTED_ALL) - assert_all_importable(utilities) diff --git a/libs/langchain/tests/unit_tests/utils/test_imports.py b/libs/langchain/tests/unit_tests/utils/test_imports.py index fdf4a957085..ea561ae4907 100644 --- a/libs/langchain/tests/unit_tests/utils/test_imports.py +++ b/libs/langchain/tests/unit_tests/utils/test_imports.py @@ -27,4 +27,3 @@ EXPECTED_ALL = [ def test_all_imports() -> None: assert set(utils.__all__) == set(EXPECTED_ALL) - assert_all_importable(utils)