sangge/BackDoorBuster
sangge/BackDoorBuster
1
0
Fork 0
Code Issues Pull Requests Projects 1 Activity

feature/cn-gpt #21

Merged
dqy merged 10 commits from feature/cn-gpt into main 2024-05-26 16:59:24 +08:00
Conversation 3 Commits 10 Files Changed 3 -15
2 changed files with 0 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit b544007e6b - Show all commits

7
tests/test_CN_GPT_detection.py
View File

@ -35,13 +35,6 @@ class TestBackdoorDetection(unittest.TestCase):
self.assertEqual(len(classified_results["medium"]), 0) self.assertEqual(len(classified_results["medium"]), 0)
self.assertEqual(len(classified_results["low"]), 0) self.assertEqual(len(classified_results["low"]), 0)
def test_gpt_env_no_set(self):
if os.getenv("BAIDU_API_KEY") is not None or os.getenv("BAIDU_SECRET_KEY") is not None:
self.skipTest("BAIDU_API_KEY or BAIDU_SECRET_KEY is set")
content = "print('test test')"
with self.assertRaises(ValueError):
detectGPT(content)
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()

8
tests/test_backdoor_detection.py
View File

@ -83,13 +83,5 @@ class TestBackdoorDetection(unittest.TestCase):
self.assertEqual(len(results["medium"]), 0) self.assertEqual(len(results["medium"]), 0)
self.assertEqual(len(results["low"]), 0) self.assertEqual(len(results["low"]), 0)
def test_gpt_env_no_set(self):
if os.getenv("OPENAI_API_KEY") is not None:
self.skipTest("OPENAI_API_KEY is setted")
content = "print('test test')"
with self.assertRaises(ValueError):
detectGPT(content)
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()