Initial import of NLProg

tests/test_command_safety.py

@@ -0,0 +1,24 @@
+import unittest
+
+from nlprog.command_safety import assess_command
+
+
+class CommandSafetyTests(unittest.TestCase):
+    def test_allows_safe_validation_command(self):
+        assessment = assess_command("python -m compileall src")
+        self.assertEqual(assessment.level, "allow")
+        self.assertFalse(assessment.blocked)
+
+    def test_confirms_package_install(self):
+        assessment = assess_command("pip install requests")
+        self.assertEqual(assessment.level, "confirm")
+        self.assertTrue(assessment.needs_confirmation)
+
+    def test_blocks_format(self):
+        assessment = assess_command("format C:")
+        self.assertEqual(assessment.level, "block")
+        self.assertTrue(assessment.blocked)
+
+
+if __name__ == "__main__":
+    unittest.main()