gistfile1.py

import itertools
import sys

import mmstats
import mmstats.reader


class FastStats(mmstats.BaseMmStats):
    a = mmstats.UInt64Field()


VALS = (0L, 9223372036854775807L, 18446744073709551615L)


def frobulate():
    stats = FastStats(path='.', filename='frobulate.mmstats')

    get = itertools.cycle(VALS).next
    print 'frobulating... %s' % stats.filename
    while 1:
        stats.a = get()


def check():
    reader = mmstats.reader.MmStatsReader.from_mmap('frobulate.mmstats')
    vals = set(VALS)
    i = 0
    errors = 0
    while 1:
        i += 1
        if i % 10000 == 0:
            print '%10d checks done: %d errors' % (i, errors)
        for stat in reader:
            if stat.value not in vals:
                print 'BOGUS VALUE: %r' % stat.value
                errors += 1
        print '.'


if __name__ == '__main__':
    if len(sys.argv) != 2:
        print 'usage: frobulate.py frob --or-- check'
    elif sys.argv[1] == 'frob':
        frobulate()
    elif sys.argv[1] == 'check':
        check()
    else:
        print 'usage: frobulate.py frob --or-- check'