示例#1
0
文件: test_all.py 项目: alepev/tofu
def test_fake_ids():

    n = 10
    fake_ids = helpers.gen_fake_ids(n)
    assert len(fake_ids) == 10

    for x in fake_ids:
        assert (x.startswith('fake'))
示例#2
0
                    help="specify output file, defaults to timestamped file.")

parser.add_argument('-H', '--human', action='store_true',
                    help="decode values into human readable format")

args = parser.parse_args()

if args.verbose:
    print(args)

TOTAL_PATIENTS = args.n

if __name__ == '__main__':

    df_output = pd.DataFrame()
    df_output['eid'] = helpers.gen_fake_ids(TOTAL_PATIENTS)

    # User has supplied field list
    if args.field is not None:
        all_field_ids = helpers.get_field_ids()
        fields_to_process = set(all_field_ids).intersection(set(args.field))
        assert len(fields_to_process) > 0, "Fields not found in lookup file."
    else:
        fields_to_process = helpers.get_field_ids()

    for field_id in tqdm(fields_to_process):
        r = helpers.get_field_metadata(field_id)

        if args.verbose is True:
            print(r)