From a7ca193dfda8d1cd53770e58da26f56545fbd6bb Mon Sep 17 00:00:00 2001
From: A Taylor <112668339+ATaylorAerospace@users.noreply.github.com>
Date: Sun, 29 Mar 2026 10:43:29 -0700
Subject: [PATCH] Add tests for legal domain classification

This file contains tests for the legal domain classification tools, ensuring that various legal texts are classified correctly according to their respective domains
---
 tests/test_tools/test_classification.py | 65 +++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 tests/test_tools/test_classification.py

diff --git a/tests/test_tools/test_classification.py b/tests/test_tools/test_classification.py
new file mode 100644
index 0000000..453dc38
--- /dev/null
+++ b/tests/test_tools/test_classification.py
@@ -0,0 +1,65 @@
+"""Tests for legal domain classification tools."""
+
+from factor.tools.classification import classify_domain
+
+
+def test_classify_contract_law():
+    """Contract-related text should classify to Contract Law."""
+    result = classify_domain(
+        provision_text="This contract involves a breach of warranty under the UCC."
+    )
+    assert "Contract" in result
+
+
+def test_classify_corporate():
+    """Corporate text should classify to Corporate/Commercial Law."""
+    result = classify_domain(
+        provision_text="The board of directors approved the merger and acquisition of securities."
+    )
+    assert "Corporate" in result
+
+
+def test_classify_ip():
+    """IP-related text should classify to Intellectual Property."""
+    result = classify_domain(
+        provision_text="Patent infringement and trademark licensing rights are at issue."
+    )
+    assert "Intellectual Property" in result
+
+
+def test_classify_environmental():
+    """Environmental text should classify to Environmental Law."""
+    result = classify_domain(
+        provision_text="EPA hazardous waste remediation and clean water compliance."
+    )
+    assert "Environmental" in result
+
+
+def test_classify_criminal():
+    """Criminal text should classify to Criminal Law."""
+    result = classify_domain(
+        provision_text="The defendant faces felony prosecution and sentencing."
+    )
+    assert "Criminal" in result
+
+
+def test_classify_tax():
+    """Tax-related text should classify to Tax Law."""
+    result = classify_domain(
+        provision_text="The IRS assessment of capital gains tax and withholding requirements."
+    )
+    assert "Tax" in result
+
+
+def test_classify_empty_defaults_to_contract():
+    """Empty text should default to Contract Law & UCC Analysis."""
+    result = classify_domain(provision_text="")
+    assert result == "Contract Law & UCC Analysis"
+
+
+def test_classify_no_keywords_defaults_to_contract():
+    """Text with no matching keywords defaults to Contract Law."""
+    result = classify_domain(
+        provision_text="The quick brown fox jumps over the lazy dog."
+    )
+    assert result == "Contract Law & UCC Analysis"