From 65e97c9b535b2e8157c62d8d0b7ba989b699c6b8 Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Thu, 8 Feb 2024 17:05:43 -0800
Subject: [PATCH] infra: mv SQLDatabase tests to community (#17276)

---
 .../tests/unit_tests/test_sql_database.py                      | 3 ++-
 .../tests/unit_tests/test_sql_database_schema.py               | 0
 2 files changed, 2 insertions(+), 1 deletion(-)
 rename libs/{langchain => community}/tests/unit_tests/test_sql_database.py (99%)
 rename libs/{langchain => community}/tests/unit_tests/test_sql_database_schema.py (100%)

diff --git a/libs/langchain/tests/unit_tests/test_sql_database.py b/libs/community/tests/unit_tests/test_sql_database.py
similarity index 99%
rename from libs/langchain/tests/unit_tests/test_sql_database.py
rename to libs/community/tests/unit_tests/test_sql_database.py
index c593faae8e8..42aaef7134f 100644
--- a/libs/langchain/tests/unit_tests/test_sql_database.py
+++ b/libs/community/tests/unit_tests/test_sql_database.py
@@ -2,7 +2,6 @@
 """Test SQL database wrapper."""
 import pytest
 import sqlalchemy as sa
-from langchain_community.utilities.sql_database import SQLDatabase, truncate_word
 from sqlalchemy import (
     Column,
     Integer,
@@ -16,6 +15,8 @@ from sqlalchemy import (
     select,
 )
 
+from langchain_community.utilities.sql_database import SQLDatabase, truncate_word
+
 metadata_obj = MetaData()
 
 user = Table(
diff --git a/libs/langchain/tests/unit_tests/test_sql_database_schema.py b/libs/community/tests/unit_tests/test_sql_database_schema.py
similarity index 100%
rename from libs/langchain/tests/unit_tests/test_sql_database_schema.py
rename to libs/community/tests/unit_tests/test_sql_database_schema.py