ilff/testgetln.py at master · aiandit/ilff · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import ilff

import sys

import random

import time

random.seed(time.time())

fname = sys.argv[1]

il = ilff.open(fname)

lines = open(fname).read().split('\n')

print('%d lines in %s' % (len(lines), fname))

print('%d lines in %s' % (il.get_nlines(), fname))

for k in range(15):
    ln = k
#    print(k, ln)
    l = il.getline(ln)
    #l = il.getlines(100, 300)
#    print(ln)
#    print(l)
#    print(type(ln))
#    print(len(lines))
#    print(lines[ln])
#    print(k, ln, l == lines[ln])
    assert(l == lines[ln] + '\n')

for k in range(100):
    ln = random.sample(range(il.get_nlines()), 1)[0]
#    print(k, ln)
    l = il.getline(ln)
    #l = il.getlines(100, 300)
#    print(ln)
#    print(l)
#    print(type(ln))
#    print(len(lines))
#    print(lines[ln])
    assert(l == lines[ln] + '\n')