Skip to content

Commit 08be70b

Browse files
author
craigsdennis
committed
Makes test a little less hardcoded
1 parent 6158206 commit 08be70b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_cells.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,6 @@ def test_referral_counts(self):
2727
"I found some referral counts that were less than 5. Check your condition.")
2828

2929
def test_total(self):
30-
self.assertEquals(len(self.df), 152,
30+
self.assertEquals(len(self.df), len(self.df.loc[(self.df.referral_count >= 5) & (self.df.email_verified == True)]),
3131
"Whoops I received a different count than I expected, make sure the last line "
3232
"is the entire resulting DataFrame (not just the head)")

0 commit comments

Comments
 (0)