def test_trajectory_too_short(self): net = ECA(30, 3) with self.assertRaises(ValueError): net.trajectory([0, 0, 0], timesteps=-1) with self.assertRaises(ValueError): net.trajectory([0, 0, 0], timesteps=0)
def test_timeseries_eca(self): for size in [5, 7, 11]: rule = ECA(30, size) time = 10 series = rule.timeseries(time) self.assertEqual((size, 2**size, time + 1), series.shape) for index, state in enumerate(rule): traj = rule.trajectory(state, timesteps=time) for t, expect in enumerate(traj): got = series[:, index, t] self.assertTrue(np.array_equal(expect, got))
def test_trajectory_eca(self): net = ECA(30, 3) with self.assertRaises(ValueError): net.trajectory([]) with self.assertRaises(ValueError): net.trajectory([0, 1]) xs = [0, 1, 0] got = net.trajectory(xs) self.assertEqual([0, 1, 0], xs) self.assertEqual([[0, 1, 0], [1, 1, 1], [0, 0, 0]], got) got = net.trajectory(xs, timesteps=5) self.assertEqual([0, 1, 0], xs) self.assertEqual([[0, 1, 0], [1, 1, 1], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]], got) got = net.trajectory(xs, encode=False) self.assertEqual([0, 1, 0], xs) self.assertEqual([[0, 1, 0], [1, 1, 1], [0, 0, 0]], got) got = net.trajectory(xs, timesteps=5, encode=False) self.assertEqual([0, 1, 0], xs) self.assertEqual([[0, 1, 0], [1, 1, 1], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]], got) got = net.trajectory(xs, encode=True) self.assertEqual([0, 1, 0], xs) self.assertEqual([2, 7, 0], list(got)) got = net.trajectory(xs, timesteps=5, encode=True) self.assertEqual([0, 1, 0], xs) self.assertEqual([2, 7, 0, 0, 0, 0], list(got)) xs = 2 got = net.trajectory(xs) self.assertEqual([2, 7, 0], list(got)) got = net.trajectory(xs, timesteps=5) self.assertEqual([2, 7, 0, 0, 0, 0], list(got)) got = net.trajectory(xs, encode=True) self.assertEqual([2, 7, 0], list(got)) got = net.trajectory(xs, timesteps=5, encode=True) self.assertEqual([2, 7, 0, 0, 0, 0], list(got)) got = net.trajectory(xs, encode=False) self.assertEqual([[0, 1, 0], [1, 1, 1], [0, 0, 0]], got) got = net.trajectory(xs, timesteps=5, encode=False) self.assertEqual([[0, 1, 0], [1, 1, 1], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]], got)