mark imports

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)

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)

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)

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)

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)

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)

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)

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

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

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)

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)

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)

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)

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)

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)

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)

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

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"

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)

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)

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)