コード例 #1
0
    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)
コード例 #2
0
    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)