def test_add_remove(self): ''' Try to remove an aggregated activity ''' loves = Love.objects.all()[:1] feed = AggregatedFeed(13) # slow version activities = [] feed.delete() for love in loves: activity = love.create_activity() activities.append(activity) feed.add(activity) assert feed.contains(activity) aggregated_activity = feed[:10][0] feed.remove(aggregated_activity) assert not feed.contains(activity)