Create test_app.py

test_app.py

@@ -0,0 +1,23 @@
+# test_app.py
+
+from app import sanitize_input, load_model, generate_text, load_json_data
+
+def test_sanitize_input():
+    assert sanitize_input("Hello, World!") == "Hello, World!"
+    assert sanitize_input("Hello@#$") == "Hello"
+    assert sanitize_input(123) == "Input must be a string."
+
+def test_load_model():
+    model, tokenizer = load_model()
+    assert model is not None
+    assert tokenizer is not None
+
+def test_load_json_data():
+    data = load_json_data()
+    assert isinstance(data, list)
+    assert "name" in data[0]
+
+def test_generate_text():
+    model, tokenizer = load_model()
+    result = generate_text(model, tokenizer, "Describe pentesting.")
+    assert isinstance(result, str)