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
|
#!/usr/bin/python3
from __future__ import print_function
import os
import sys
sys.path.insert(0, 'src')
sys.path.insert(0, 'build/src')
import plfit
prefixes = [ "data", "../data", "../../data" ]
if 'DATADIR' in os.environ:
prefixes.append(os.environ.get('DATADIR'))
DATA_DIR=None
for prefix in prefixes:
if os.path.isdir(prefix):
DATA_DIR=prefix
break
if not os.path.isdir(DATA_DIR):
print("Can't find the data directory!", file=sys.stderr)
sys.exit(1)
print("Testing continuous data...")
data = [float(line) for line in open(os.path.join(DATA_DIR, "continuous_data_small.txt"))]
result = plfit.plfit_continuous(data)
print("alpha =", result.alpha)
print("xmin =", result.xmin)
print("L =", result.L)
print()
print("Testing discrete data...")
data = [float(line) for line in open(os.path.join(DATA_DIR, "discrete_data.txt"))]
result = plfit.plfit_discrete(data)
print("alpha =", result.alpha)
print("xmin =", result.xmin)
print("L =", result.L)
|
