Forráskód Böngészése

Remove extra object wrap around json, update json test.

yk396 4 éve
szülő
commit
ec24ffa4c3
2 módosított fájl, 3 hozzáadás és 4 törlés
  1. 2 3
      howdoi/howdoi.py
  2. 1 1
      test_howdoi.py

+ 2 - 3
howdoi/howdoi.py

@@ -369,8 +369,7 @@ def _get_instructions(args):
         answer += '\n'
         answers.append({"answer": answer, "link": link, "position": current_position})
 
-    res = {}
-    res["answers"] = answers
+    res = answers
 
     return json.dumps(res)
 
@@ -406,7 +405,7 @@ def _parse_json(res, args):
     answer_spliter = '\n' + '=' * spliter_length + '\n\n'
 
     formated_answers = []
-    for answer in res["answers"]:
+    for answer in res:
 
         next_ans = answer["answer"]
         if args["link"]: #  if we only want links

+ 1 - 1
test_howdoi.py

@@ -109,7 +109,7 @@ class HowdoiTestCase(unittest.TestCase):
         json_answer = json.loads(json_answer)
         print(json_answer)
         print(txt_answer)
-        self.assertEqual(json_answer["answers"]["answer"], txt_answer)
+        self.assertEqual(json_answer[0]["answer"], txt_answer)
 
     def test_multiple_answers(self):
         query = self.queries[0]