def test_particles_observable(self):
fname = os.path.join(self.dir, "test_observables_particles.h5")
context = Context()
context.box_size = [13., 13., 13.]
context.particle_types.add("A", .1)
context.particle_types.add("B", .1)
sim = Simulation("SingleCPU", context)
sim.add_particle("A", common.Vec(0, 0, 0))
sim.add_particle("B", common.Vec(0, 0, 0))
# every time step, add one particle
sim.register_observable_n_particles(
1, ["A"],
lambda n: sim.add_particle("A", common.Vec(1.5, 2.5, 3.5)))
handle = sim.register_observable_particles(1)
n_timesteps = 19
with closing(io.File.create(fname)) as f:
handle.enable_write_to_file(f, u"particles", int(3))
loop = sim.create_loop(0)
loop.write_config_to_file(f)
loop.run(n_timesteps)
handle.flush()
from readdy.util.io_utils import get_particle_types
particle_types = get_particle_types(fname)
with h5py.File(fname, "r") as f2:
types = f2["readdy/observables/particles/types"][:]
ids = f2["readdy/observables/particles/ids"][:]
positions = f2["readdy/observables/particles/positions"][:]
for t in range(n_timesteps):
np.testing.assert_equal(len(types[t]), t + 3)
np.testing.assert_equal(len(ids[t]), t + 3)
np.testing.assert_equal(len(positions[t]), t + 3)
np.testing.assert_equal(types[t][0],
particle_types["A"]["type_id"])
np.testing.assert_equal(positions[t][0][0], 0)
np.testing.assert_equal(positions[t][0][1], 0)
np.testing.assert_equal(positions[t][0][2], 0)
np.testing.assert_equal(positions[t][1][0], 0)
np.testing.assert_equal(positions[t][1][1], 0)
np.testing.assert_equal(positions[t][1][2], 0)
np.testing.assert_equal(types[t][1],
particle_types["B"]["type_id"])
for others in range(2, len(types[t])):
np.testing.assert_equal(types[t][others],
particle_types["A"]["type_id"])
np.testing.assert_equal(positions[t][others][0], 1.5)
np.testing.assert_equal(positions[t][others][1], 2.5)
np.testing.assert_equal(positions[t][others][2], 3.5)
def test_particles_observable(self):
fname = os.path.join(self.dir, "test_observables_particles.h5")
sim = Simulation("SingleCPU")
sim.context.box_size = [13.,13.,13.]
sim.context.particle_types.add("A", .1)
sim.context.particle_types.add("B", .1)
sim.add_particle("A", common.Vec(0, 0, 0))
sim.add_particle("B", common.Vec(0, 0, 0))
# every time step, add one particle
sim.register_observable_n_particles(1, ["A"], lambda n: sim.add_particle("A", common.Vec(1.5, 2.5, 3.5)))
handle = sim.register_observable_particles(1)
n_timesteps = 19
with closing(io.File.create(fname)) as f:
handle.enable_write_to_file(f, u"particles", int(3))
loop = sim.create_loop(0)
loop.write_config_to_file(f)
loop.run(n_timesteps)
handle.flush()
from readdy.util.io_utils import get_particle_types
particle_types = get_particle_types(fname)
with h5py.File(fname, "r") as f2:
types = f2["readdy/observables/particles/types"][:]
ids = f2["readdy/observables/particles/ids"][:]
positions = f2["readdy/observables/particles/positions"][:]
for t in range(n_timesteps):
np.testing.assert_equal(len(types[t]), t + 3)
np.testing.assert_equal(len(ids[t]), t + 3)
np.testing.assert_equal(len(positions[t]), t + 3)
np.testing.assert_equal(types[t][0], particle_types["A"]["type_id"])
np.testing.assert_equal(positions[t][0][0], 0)
np.testing.assert_equal(positions[t][0][1], 0)
np.testing.assert_equal(positions[t][0][2], 0)
np.testing.assert_equal(positions[t][1][0], 0)
np.testing.assert_equal(positions[t][1][1], 0)
np.testing.assert_equal(positions[t][1][2], 0)
np.testing.assert_equal(types[t][1], particle_types["B"]["type_id"])
for others in range(2, len(types[t])):
np.testing.assert_equal(types[t][others], particle_types["A"]["type_id"])
np.testing.assert_equal(positions[t][others][0], 1.5)
np.testing.assert_equal(positions[t][others][1], 2.5)
np.testing.assert_equal(positions[t][others][2], 3.5)
def test_particles_observable(self):
fname = os.path.join(self.dir, "test_observables_particles.h5")
sim = Simulation()
sim.set_kernel("SingleCPU")
sim.box_size = common.Vec(13, 13, 13)
typeid_A = sim.register_particle_type("A", .1, .1)
typeid_B = sim.register_particle_type("B", .1, .1)
sim.add_particle("A", common.Vec(0, 0, 0))
sim.add_particle("B", common.Vec(0, 0, 0))
# every time step, add one particle
sim.register_observable_n_particles(
1, ["A"],
lambda n: sim.add_particle("A", common.Vec(1.5, 2.5, 3.5)))
handle = sim.register_observable_particles(1)
n_timesteps = 19
with closing(
io.File(fname, io.FileAction.CREATE,
io.FileFlag.OVERWRITE)) as f:
handle.enable_write_to_file(f, u"particles", int(3))
sim.run_scheme_readdy(True).configure(0).run(n_timesteps)
handle.flush()
with h5py.File(fname, "r") as f2:
types = f2["readdy/observables/particles/types"][:]
ids = f2["readdy/observables/particles/ids"][:]
positions = f2["readdy/observables/particles/positions"][:]
for t in range(n_timesteps):
np.testing.assert_equal(len(types[t]), t + 3)
np.testing.assert_equal(len(ids[t]), t + 3)
np.testing.assert_equal(len(positions[t]), t + 3)
np.testing.assert_equal(types[t][0], typeid_A)
np.testing.assert_equal(positions[t][0][0], 0)
np.testing.assert_equal(positions[t][0][1], 0)
np.testing.assert_equal(positions[t][0][2], 0)
np.testing.assert_equal(positions[t][1][0], 0)
np.testing.assert_equal(positions[t][1][1], 0)
np.testing.assert_equal(positions[t][1][2], 0)
np.testing.assert_equal(types[t][1], typeid_B)
for others in range(2, len(types[t])):
np.testing.assert_equal(types[t][others], typeid_A)
np.testing.assert_equal(positions[t][others][0], 1.5)
np.testing.assert_equal(positions[t][others][1], 2.5)
np.testing.assert_equal(positions[t][others][2], 3.5)
def test_particles_observable(self):
fname = os.path.join(self.dir, "test_observables_particles.h5")
sim = Simulation()
sim.set_kernel("SingleCPU")
sim.box_size = common.Vec(13, 13, 13)
typeid_A = sim.register_particle_type("A", .1)
typeid_B = sim.register_particle_type("B", .1)
sim.add_particle("A", common.Vec(0, 0, 0))
sim.add_particle("B", common.Vec(0, 0, 0))
# every time step, add one particle
sim.register_observable_n_particles(1, ["A"], lambda n: sim.add_particle("A", common.Vec(1.5, 2.5, 3.5)))
handle = sim.register_observable_particles(1)
n_timesteps = 19
with closing(io.File.create(fname)) as f:
handle.enable_write_to_file(f, u"particles", int(3))
sim.run_scheme_readdy(True).configure(0).run(n_timesteps)
handle.flush()
with h5py.File(fname, "r") as f2:
types = f2["readdy/observables/particles/types"][:]
ids = f2["readdy/observables/particles/ids"][:]
positions = f2["readdy/observables/particles/positions"][:]
for t in range(n_timesteps):
np.testing.assert_equal(len(types[t]), t + 3)
np.testing.assert_equal(len(ids[t]), t + 3)
np.testing.assert_equal(len(positions[t]), t + 3)
np.testing.assert_equal(types[t][0], typeid_A)
np.testing.assert_equal(positions[t][0][0], 0)
np.testing.assert_equal(positions[t][0][1], 0)
np.testing.assert_equal(positions[t][0][2], 0)
np.testing.assert_equal(positions[t][1][0], 0)
np.testing.assert_equal(positions[t][1][1], 0)
np.testing.assert_equal(positions[t][1][2], 0)
np.testing.assert_equal(types[t][1], typeid_B)
for others in range(2, len(types[t])):
np.testing.assert_equal(types[t][others], typeid_A)
np.testing.assert_equal(positions[t][others][0], 1.5)
np.testing.assert_equal(positions[t][others][1], 2.5)
np.testing.assert_equal(positions[t][others][2], 3.5)
