Example #1
0
@author: jntrcs
"""

import pickle
from OuterGraph import OuterGraph
from Widget import Widget

# Step 2
with open('vertice_ints', 'rb') as file:

    # Step 3
    vertices = pickle.load(file)

# Step 2
with open('int_to_states', 'rb') as file:

    # Step 3
    key = pickle.load(file)

graph = OuterGraph(key, vertices)
test_widget = Widget(1, key[1], graph)
evs = test_widget.generate_expected_values()

for k, vals in vertices.items():
    if 15375 in vals:
        print(k)

print(key[138830])

bw = Widget(138830, key[138830], graph)
evs = bw.get_expected_values()