|
7 | 7 | # add_cats |
8 | 8 | ########################################################################### |
9 | 9 |
|
10 | | -def test__add_cats__succeeds(): |
11 | | - c = cattery.Cattery() |
12 | | - c.add_cats(["Fluffy", "Snookums"]) |
13 | | - assert c.cats == ["Fluffy", "Snookums"] |
14 | | - assert c.num_cats == 2 |
| 10 | +@pytest.fixture() |
| 11 | +def cattery_client(): |
| 12 | + return cattery.Cattery() |
| 13 | + |
| 14 | + |
| 15 | +def test__add_cats__succeeds(cattery_client): |
| 16 | + cattery_client.add_cats(["Fluffy", "Snookums"]) |
| 17 | + assert cattery_client.cats == ["Fluffy", "Snookums"] |
| 18 | + assert cattery_client.num_cats == 2 |
15 | 19 |
|
16 | 20 |
|
17 | 21 | ########################################################################### |
18 | 22 | # remove_cat |
19 | 23 | ########################################################################### |
20 | 24 |
|
21 | | -def test__remove_cat__succeeds(): |
22 | | - c = cattery.Cattery() |
23 | | - c.add_cats(["Fluffy", "Junior"]) |
24 | | - c.remove_cat("Fluffy") |
25 | | - assert c.cats == ["Junior"] |
26 | | - assert c.num_cats == 1 |
| 25 | +def test__remove_cat__succeeds(cattery_client): |
| 26 | + cattery_client.add_cats(["Fluffy", "Junior"]) |
| 27 | + cattery_client.remove_cat("Fluffy") |
| 28 | + assert cattery_client.cats == ["Junior"] |
| 29 | + assert cattery_client.num_cats == 1 |
27 | 30 |
|
28 | 31 |
|
29 | | -def test__remove_cat__no_cats__fails(): |
30 | | - c = cattery.Cattery() |
| 32 | +def test__remove_cat__no_cats__fails(cattery_client): |
31 | 33 | with pytest.raises(cattery.CatNotFound): |
32 | | - c.remove_cat("Fluffles") |
| 34 | + cattery_client.remove_cat("Fluffles") |
33 | 35 |
|
34 | 36 |
|
35 | | -def test__remove_cat__cat_not_in_cattery__fails(): |
36 | | - c = cattery.Cattery() |
37 | | - c.add_cats(["Fluffy"]) |
| 37 | +def test__remove_cat__cat_not_in_cattery__fails(cattery_client): |
| 38 | + cattery_client.add_cats(["Fluffy"]) |
38 | 39 | with pytest.raises(cattery.CatNotFound): |
39 | | - c.remove_cat("Snookums") |
| 40 | + cattery_client.remove_cat("Snookums") |
0 commit comments