Refactored recently merged in tests

cms/djangoapps/contentstore/features/common.py

@@ -233,7 +233,7 @@ def shows_captions(step, show_captions):
 def save_button_disabled(step):
     button_css = '.action-save'
     disabled = 'is-disabled'
-    assert world.css_find(button_css)[0].has_class(disabled)
+    assert world.css_has_class(button_css, disabled)
 
 
 def type_in_codemirror(index, text):

cms/djangoapps/contentstore/features/grading.py

@@ -104,8 +104,7 @@ def populate_course(step):
 def changes_not_persisted(step):
     reload_the_page(step)
     name_id = '#course-grading-assignment-name'
-    ele = world.css_find(name_id)[0]
-    assert(ele.value == 'Homework')
+    assert(world.css_value(name_id) == 'Homework')
 
 
 @step(u'I see the assignment type "(.*)"$')