def test_network_export(self, user_data, friends, posts_data): user = User(data=user_data) friends[0].update_posts(data=posts_data) friends[1].update_posts(data=posts_data) user.update_friends(friends=friends) path = user.walk() export_file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'test_network_export' ) data = user.export(export_file_path)
def test_walk(self, user_data, friends, posts_data): user = User(data=user_data) friends[0].update_posts(data=posts_data) friends[1].update_posts(data=posts_data) user.update_friends(friends=friends) path = user.walk() assert len(path) == 6 assert isinstance(path[0], User) assert isinstance(path[1], Post) assert isinstance(path[2], Post) assert isinstance(path[3], User) assert isinstance(path[4], Post) assert isinstance(path[5], Post)