Use parametrized tests where it possible

tests/rules/test_rm_root.py

@@ -1,18 +1,21 @@
-from mock import Mock
+import pytest
 from thefuck.rules.rm_root import match, get_new_command
+from tests.utils import Command
 
 
 def test_match():
-    assert match(Mock(script='rm -rf /',
-                      stderr='add --no-preserve-root'), None)
-    assert not match(Mock(script='ls',
-                          stderr='add --no-preserve-root'), None)
-    assert not match(Mock(script='rm --no-preserve-root /',
-                          stderr='add --no-preserve-root'), None)
-    assert not match(Mock(script='rm -rf /',
-                          stderr=''), None)
+    assert match(Command(script='rm -rf /',
+                         stderr='add --no-preserve-root'), None)
+
+
+@pytest.mark.parametrize('command', [
+    Command(script='ls', stderr='add --no-preserve-root'),
+    Command(script='rm --no-preserve-root /', stderr='add --no-preserve-root'),
+    Command(script='rm -rf /', stderr='')])
+def test_not_match(command):
+    assert not match(command, None)
 
 
 def test_get_new_command():
-    assert get_new_command(Mock(script='rm -rf /'), None) \
-        == 'rm -rf / --no-preserve-root'
+    assert get_new_command(Command(script='rm -rf /'), None) \
+           == 'rm -rf / --no-preserve-root'