agent-governance-toolkit/packages/agent-os/tests

Author: yacinemebarki

packages/agent-os/src/agent_os/policies/cli.py

@@ -44,6 +44,7 @@ def policy_violation(msg):
 
 def passed_check(msg):
     print(f"[green]✔ {msg}[/green]", file=sys.stderr)
+    
 
 # ---------------------------------------------------------------------------
 # Lazy imports

packages/agent-os/src/test_policies/test_policy_messages.yaml

@@ -0,0 +1,25 @@
+# test_cli_run.py
+from agent_os.policies.cli import success, error, warn, policy_violation, passed_check
+
+def main():
+    print("=== Testing CLI functions with colored output ===\n")
+
+    print("1️⃣  success():")
+    success("This is a success message")
+
+    print("\n2️⃣  error():")
+    error("This is an error message")
+
+    print("\n3️⃣  warn():")
+    warn("This is a warning message")
+
+    print("\n4️⃣  policy_violation():")
+    policy_violation("Policy violation detected!")
+
+    print("\n5️⃣  passed_check():")
+    passed_check("Test passed successfully!")
+
+    print("\n=== Done testing CLI functions ===")
+
+if __name__ == "__main__":
+    main()