Python collect_user_information Examples

Example #1

File: github_util.py Project: xiaohan2012/graph-event-detection

def dump2people(input_path, output_path):
    df = make_dataframe(input_path)
    user_info = collect_user_information(
        df,
        id_field='sender_id',
        other_fields=[]
    )
    user_info.rename(columns={'sender_id': 'id'}, inplace=True)
    user_info.to_json(output_path, orient="records")

Example #2

File: test_threads_util.py Project: xiaohan2012/lst

def test_collect_user_information():
    data = pd.DataFrame(columns=['sender_id', 'sender_name'],
                        data={'sender_id': [1, 2, 3, 1, 2],
                              'sender_name': ['One', 'Two', 'Three',
                                              'One', 'Two']})
    actual = collect_user_information(
        data,
        id_field='sender_id',
        other_fields=['sender_name']
    )
    expected = data[:3]
    assert_true(expected.equals(actual))

Example #3

File: test_threads_util.py Project: xiaohan2012/graph-event-detection

def test_collect_user_information():
    data = pd.DataFrame(columns=['sender_id', 'sender_name'],
                        data={
                            'sender_id': [1, 2, 3, 1, 2],
                            'sender_name':
                            ['One', 'Two', 'Three', 'One', 'Two']
                        })
    actual = collect_user_information(data,
                                      id_field='sender_id',
                                      other_fields=['sender_name'])
    expected = data[:3]
    assert_true(expected.equals(actual))