CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-08-10 19:50:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(py upgrade):

Browse Source
This commit is contained in:
Alexander Myasoedov
2026-05-14 18:56:24 +03:00
parent 46ec775266
commit a0b2b9ec70
14 changed files with 681 additions and 558 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/attack_rules/test_dataset.py
+4 -8
View File
@@ -56,20 +56,16 @@ class TestRulesToDataset:
class TestLoadRulesAsDataset:
def test_basic_load(self):
with tempfile.TemporaryDirectory() as tmpdir:
(Path(tmpdir) / "rule1.yaml").write_text(
"""
(Path(tmpdir) / "rule1.yaml").write_text("""
name: test1
type: jailbreak
prompt: Jailbreak prompt
"""
)
(Path(tmpdir) / "rule2.yaml").write_text(
"""
""")
(Path(tmpdir) / "rule2.yaml").write_text("""
name: test2
type: harmful
prompt: Harmful prompt
"""
)
""")
dataset = load_rules_as_dataset(tmpdir)
assert len(dataset.prompts) == 2