test: add GitHub action for unit test (#24)

Author: Mini256

.github/workflows/test.yml

@@ -0,0 +1,30 @@
+name: Test
+
+on:
+  push:
+    branches:
+      - main
+  pull_request_target:
+    branches:
+      - main
+
+jobs:
+  test:
+    name: Test
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+      - name: Install uv
+        uses: astral-sh/setup-uv@v5
+        with:
+          enable-cache: true
+      - name: Install dependencies
+        run: uv sync --all-extras --dev
+      - name: Run Lint
+        run: uv tool run ruff check
+      - name: Run Format
+        run: uv tool run ruff format
+      - name: Run Test
+        run: uv run pytest tests
+

examples/text2sql/main.py

@@ -67,9 +67,7 @@ class QuestionSQLResponse(BaseModel):
 table_definitions = []
 current_database = db._db_engine.url.database
 for table_name in db.table_names():
-    table_definitions.append(
-        db.query(f"SHOW CREATE TABLE `{table_name}`").to_rows()[0]
-    )
+    table_definitions.append(db.query(f"SHOW CREATE TABLE `{table_name}`").to_rows()[0])
 
 
 def on_submit():
@@ -98,7 +96,7 @@ def on_submit():
         )
         st.session_state.past.append(user_input)
 
-        if 'insert' in response.sql.lower() or 'update' in response.sql.lower():
+        if "insert" in response.sql.lower() or "update" in response.sql.lower():
             st.error(
                 "The generated SQL is not a SELECT statement, please check it carefully before running it."
             )