def test_nestedlife(testdata, func):
model = nestedlife.build(load_saved=False)
data = get_nested(func(model), 'PolsSurr')
with open(testdata, 'rb') as file:
data_saved = pickle.load(file)
assert data == data_saved
def test_nestedlife():
model = nestedlife.build(load_saved=False)
for testdata, func in zip([testdata1, testdata2],
[set_model, update_model]):
data = get_nested(func(model), 'PolsSurr')
with open(testdata, 'rb') as file:
data_saved = pickle.load(file)
assert data == data_saved
if t < t0:
expect_t0[t] = 0
else:
cells = outer.InnerProjection[t0].cells[item]
expect_t0[t] = cells[t]
expect.append(expect_t0)
act = [round_signif(val, 10) for val in act]
expect = [[round_signif(val, 10) for val in inner] for inner in expect]
return [act, expect]
def save_data(outer, filename):
data = get_nested(outer, 'nop_Surrender')
filepath = os.path.join(os.path.dirname(__file__), filename)
with open(filepath, 'wb') as file:
pickle.dump(data, file, protocol=4)
def generate_data(model):
save_data(set_model(model), 'data_nestedlife1')
save_data(update_model(model), 'data_nestedlife2')
if __name__ == '__main__':
generate_data(model=nestedlife.build(True))