From b4e09341b354d12d419977eb5907d310eb3d226a Mon Sep 17 00:00:00 2001
From: Arnaud Gelas <arnaudgelas@gmail.com>
Date: Thu, 18 Jan 2024 20:47:33 +0100
Subject: [PATCH] Stop generating unit test for non python files

When trying to create a simple HelloWorld with test, metagpt creates
test for README.md
---
 metagpt/roles/qa_engineer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/metagpt/roles/qa_engineer.py b/metagpt/roles/qa_engineer.py
index 0e323893e..45a1c7715 100644
--- a/metagpt/roles/qa_engineer.py
+++ b/metagpt/roles/qa_engineer.py
@@ -65,6 +65,8 @@ class QaEngineer(Role):
             code_doc = await src_file_repo.get(filename)
             if not code_doc:
                 continue
+            if not code_doc.filename.endswith(".py"):
+                continue
             test_doc = await tests_file_repo.get("test_" + code_doc.filename)
             if not test_doc:
                 test_doc = Document(