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

fix：删除无用测试代码——api_key

Browse Source
This commit is contained in:
ccyj 2024-05-24 20:44:35 +08:00
parent b1bc566c09
commit b544007e6b
2 changed files with 0 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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()