def test_sum_stats_save_load(history: History):
arr = sp.random.rand(10)
arr2 = sp.random.rand(10, 2)
particle_list = [
Particle(0, Parameter({
"a": 23,
"b": 12
}), .2, [.1], [{
"ss1": .1,
"ss2": arr2,
"ss3": example_df(),
"rdf0": r["iris"]
}], [], True),
Particle(0, Parameter({
"a": 23,
"b": 12
}), .2, [.1], [{
"ss12": .11,
"ss22": arr,
"ss33": example_df(),
"rdf": r["mtcars"]
}], [], True)
]
history.append_population(0, 42, Population(particle_list), 2,
["m1", "m2"])
weights, sum_stats = history.get_sum_stats(0, 0)
assert (weights == 0.5).all()
assert sum_stats[0]["ss1"] == .1
assert (sum_stats[0]["ss2"] == arr2).all()
assert (sum_stats[0]["ss3"] == example_df()).all().all()
assert (sum_stats[0]["rdf0"] == pandas2ri.ri2py(r["iris"])).all().all()
assert sum_stats[1]["ss12"] == .11
assert (sum_stats[1]["ss22"] == arr).all()
assert (sum_stats[1]["ss33"] == example_df()).all().all()
assert (sum_stats[1]["rdf"] == pandas2ri.ri2py(r["mtcars"])).all().all()
def test_sum_stats_save_load(history: History):
import scipy as sp
arr = sp.random.rand(10)
arr2 = sp.random.rand(10, 2)
particle_population = [
ValidParticle(0, Parameter({
"a": 23,
"b": 12
}), .2, [.1], [{
"ss1": .1,
"ss2": arr2
}]),
ValidParticle(0, Parameter({
"a": 23,
"b": 12
}), .2, [.1], [{
"ss12": .11,
"ss22": arr
}])
]
history.append_population(0, 42, particle_population, 2, ["m1", "m2"])
weights, sum_stats = history.get_sum_stats(0, 0)
assert (weights == 0.5).all()
assert sum_stats[0]["ss1"] == .1
assert (sum_stats[0]["ss2"] == arr2).all()
assert sum_stats[1]["ss12"] == .11
assert (sum_stats[1]["ss22"] == arr).all()
