Skip to content

Commit 9003491

Browse files
committed
Replace assert with assertEqual statements
1 parent 45ba1a1 commit 9003491

File tree

1 file changed

+10
-10
lines changed
  • stream_framework/tests/feeds/notification_feed

1 file changed

+10
-10
lines changed

stream_framework/tests/feeds/notification_feed/base.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ def setUp(self):
4646
self.love_id = self.aggregated_love.serialization_id
4747

4848
def assert_activity_markers(self, aggregated_activity, seen=False, read=False):
49-
assert aggregated_activity.is_seen == seen
50-
assert aggregated_activity.is_read == read
49+
self.assertEqual(aggregated_activity.is_seen, seen)
50+
self.assertEqual(aggregated_activity.is_read, read)
5151

5252
def assert_activities_markers(self, aggregated_activities, seen=False, read=False):
5353
for aggregated_activity in aggregated_activities:
@@ -82,8 +82,8 @@ def test_remove_activities(self):
8282
self.test_feed.remove_many(self.loves)
8383

8484
self.assertEqual(self.test_feed[:], [])
85-
assert self.test_feed.count_unseen() == 0
86-
assert self.test_feed.count_unread() == 0
85+
self.assertEqual(self.test_feed.count_unseen(), 0)
86+
self.assertEqual(self.test_feed.count_unread(), 0)
8787

8888
@implementation
8989
def test_remove_many_aggregated_activities(self):
@@ -95,8 +95,8 @@ def test_remove_many_aggregated_activities(self):
9595

9696
self.test_feed.remove_many_aggregated([self.aggregated_comment, self.aggregated_love])
9797
self.assertEqual(self.test_feed[:], [])
98-
assert self.test_feed.count_unseen() == 0
99-
assert self.test_feed.count_unread() == 0
98+
self.assertEqual(self.test_feed.count_unseen(), 0)
99+
self.assertEqual(self.test_feed.count_unread(), 0)
100100

101101
@implementation
102102
def test_mark_aggregated_activity(self):
@@ -148,9 +148,9 @@ def test_mark_all_aggreagted_activities_as_read(self):
148148
@implementation
149149
def test_delete_feed(self):
150150
self.test_feed.add_many(self.loves)
151-
assert self.test_feed.count_unseen() == 1
152-
assert self.test_feed.count_unread() == 1
151+
self.assertEqual(self.test_feed.count_unseen(), 1)
152+
self.assertEqual(self.test_feed.count_unread(), 1)
153153

154154
self.test_feed.delete()
155-
assert self.test_feed.count_unseen() == 0
156-
assert self.test_feed.count_unread() == 0
155+
self.assertEqual(self.test_feed.count_unseen(), 0)
156+
self.assertEqual(self.test_feed.count_unread(), 0)

0 commit comments

Comments
 (0)