test_network.py 文件源码

def setUp(self):
        self.close_friends = set(["David", "Christy", "Carmela", "Evelyn"])
        self.not_so_close_friends = set(["Tony", "Ivan", "Alex", "Julie"])
        self.strangers = set(["UnknownGuy", "UnknownGirl"])
        self.dislike = set(["Marcus"])

        self.Group = namedtuple("Data", ["group_name", "value"])
        self.Data = namedtuple("Errors", ["error", "group"])
        self.new_group = self.Group("haters", set(["Willock"]))
        self.existing_group = self.Group("strangers", set(["J.Cole"]))
        self.bad_value_group = self.Group("love", None)
        self.bad_groups = [self.Data(NameError, self.existing_group),
                           self.Data(TypeError, self.bad_value_group)]

        self.a_group_name = "strangers"
        self.not_a_group_name = "power_rangers"

        self.network = Network(self.close_friends, 
                               self.not_so_close_friends,
                               self.strangers,
                               self.dislike)

        self.print_network_lines = ["Social Network:",
                                    "close_friends: {",
                                    "people_i_dislike: {'Marcus'}",
                                    "not_so_close_friends: {",
                                    "strangers: {"]