From ebac62e21ac1734dfd41b1e3207ebec1c3107060 Mon Sep 17 00:00:00 2001
From: doa <dogakantarci@gmail.com>
Date: Sat, 8 Mar 2025 19:01:57 +0300
Subject: [PATCH] fix: issue #156 - move test file to tests dir

---
 .gitignore                                                   | 1 +
 agentic_security/probe_actor/refusal.py                      | 2 +-
 {agentic_security => tests}/refusal_classifier/test_model.py | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)
 rename {agentic_security => tests}/refusal_classifier/test_model.py (95%)

diff --git a/.gitignore b/.gitignore
index 8524ead..d7016a2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -17,3 +17,4 @@ inv/
 scripts/
 docx/
 agentic_security.toml
+/venv
\ No newline at end of file
diff --git a/agentic_security/probe_actor/refusal.py b/agentic_security/probe_actor/refusal.py
index a135423..2d5b077 100644
--- a/agentic_security/probe_actor/refusal.py
+++ b/agentic_security/probe_actor/refusal.py
@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 
-from agentic_security.refusal_classifier import RefusalClassifier
+from agentic_security.refusal_classifier.model import RefusalClassifier
 
 classifier = RefusalClassifier()
 classifier.load_model()
diff --git a/agentic_security/refusal_classifier/test_model.py b/tests/refusal_classifier/test_model.py
similarity index 95%
rename from agentic_security/refusal_classifier/test_model.py
rename to tests/refusal_classifier/test_model.py
index b1f017b..1a4677b 100644
--- a/agentic_security/refusal_classifier/test_model.py
+++ b/tests/refusal_classifier/test_model.py
@@ -3,7 +3,7 @@ from unittest.mock import patch
 import pandas as pd
 import pytest
 
-from .model import RefusalClassifier
+from agentic_security.refusal_classifier.model import RefusalClassifier
 
 
 @pytest.fixture