added weights

mitodl · Apr 4, 2018 · d812e38 · d812e38
1 parent c99fa9b
commit d812e38
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/edx_sga/tests/integration_tests.py b/edx_sga/tests/integration_tests.py
@@ -112,6 +112,7 @@ def make_one(self, display_name=None, **kw):
         """
         Creates a XBlock SGA for testing purpose.
         """
+        weight = kw.get('weight', 0)
         field_data = DictFieldData(kw)
         block = StaffGradedAssignmentXBlock(self.runtime, field_data, self.scope_ids)
         block.location = Location(
@@ -126,7 +127,7 @@ def make_one(self, display_name=None, **kw):
             block.display_name = display_name
 
         block.start = datetime.datetime(2010, 5, 12, 2, 42, tzinfo=pytz.utc)
-        block.weight = field_data.get('weight', 0)
+        block.weight = weight
         modulestore().create_item(
             self.staff.username, block.location.course_key, block.location.block_type, block.location.block_id
         )