Add test_ht and test_e_diagnosis

phamquiluan · Jan 10, 2025 · 90a24ee · 90a24ee
1 parent 45c7339
commit 90a24ee
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/test_rcd.py b/tests/test_rcd.py
@@ -18,3 +18,19 @@ def test_rcd():
 
     # Check if the script ran successfully
     assert result.returncode == 0, f"Script failed with return code {result.returncode}\nOutput: {result.stdout}\nError: {result.stderr}"
+
+
+def test_ht():
+    command = ["python", "main-ase.py", "--method", "ht", "--dataset", "online-boutique", "--test"]
+    result = subprocess.run(command, capture_output=True, text=True)
+
+    # Check if the script ran successfully
+    assert result.returncode == 0, f"Script failed with return code {result.returncode}\nOutput: {result.stdout}\nError: {result.stderr}"
+
+
+def test_e_diagnosis():
+    command = ["python", "main-ase.py", "--method", "e_diagnosis", "--dataset", "online-boutique", "--test"]
+    result = subprocess.run(command, capture_output=True, text=True)
+
+    # Check if the script ran successfully
+    assert result.returncode == 0, f"Script failed with return code {result.returncode}\nOutput: {result.stdout}\nError: {result.stderr}"