def test_users_related(self): users = db_update.users_related(2) # unfollowは対象外 # 2 <- 4(unfollow) # 1 <- 2 users.sort(cmp = lambda a,b: cmp(a.user_id, b.user_id)) self.assertEquals(len(users), 2) self.assertEquals(users[0].user_id, 1) self.assertEquals(users[1].user_id, 2)
def test_users_related_all(self): users = db_update.users_related(2, contains_unfollow = True) # 2 <- 4(unfollow) # 1 <- 2 users.sort(cmp = lambda a,b: cmp(a.user_id, b.user_id)) self.assertEquals(len(users), 3) self.assertEquals(users[0].user_id, 1) self.assertEquals(users[1].user_id, 2) self.assertEquals(users[2].user_id, 4)