From 3966ae20fc68041721f97c45ff92102ed5729aee Mon Sep 17 00:00:00 2001
From: ankitjavalkar
Date: Sat, 16 Jan 2021 23:42:30 +0530
Subject: Fix failing test case

---
 upload/utils.py | 2 +-
 yaksh/models.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/upload/utils.py b/upload/utils.py
index 3609f09..cae9f1f 100644
--- a/upload/utils.py
+++ b/upload/utils.py
@@ -255,7 +255,7 @@ def create_header(data, dtype):
         "type": dtype,
         "data": data
     }
-    yaml=YAML()
+    yaml=ruamel.yaml.YAML()
     yaml.default_flow_style = False
     io_obj = io.StringIO()
 
diff --git a/yaksh/models.py b/yaksh/models.py
index bdac927..00bca5b 100644
--- a/yaksh/models.py
+++ b/yaksh/models.py
@@ -2940,7 +2940,7 @@ class TOCManager(models.Manager):
                         else:
                             que = Question.objects.create(**content)
                             for test_case in test_cases:
-                                test_case_type = test_case.pop('type')
+                                test_case_type = test_case.pop('test_case_type')
                                 model_class = get_model_class(test_case_type)
                                 model_class.objects.get_or_create(
                                     question=que, **test_case, type=test_case_type
-- 
cgit