add parametrized test

brownie-mix · Oct 2, 2019 · 7654173 · 7654173
1 parent f6592b6
commit 7654173
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/test_approve_transferFrom.py b/tests/test_approve_transferFrom.py
@@ -28,10 +28,8 @@ def test_transferFrom(token, accounts):
     assert token.allowance(accounts[0], accounts[1]) == "1 ether"
 
 
-def test_transferFrom_reverts(token, accounts):
+@pytest.mark.parametrize('idx', [0, 1, 2])
+def test_transferFrom_reverts(token, accounts, idx):
     '''transerFrom should revert'''
     with pytest.reverts("Insufficient allowance"):
-        token.transferFrom(accounts[0], accounts[2], "1 ether", {'from': accounts[1]})
-
-    with pytest.reverts("Insufficient allowance"):
-        token.transferFrom(accounts[0], accounts[2], "1 ether", {'from': accounts[0]})
+        token.transferFrom(accounts[0], accounts[2], "1 ether", {'from': accounts[idx]})