def test_nans(self):
# Spikes in 1/4 of the arena, uniform occupancy
rate_map = np.array([10., 0., 0., np.nan])
px = np.array([.25, .25, .25, .25])
rate = information_rate(rate_map, px)
spec = information_specificity(rate_map, px)
assert rate == 5 # bits/s
assert spec == 2 # bits/spike
def test_basic(self):
# Spikes in half of the arena, uniform occupancy
rate_map = np.array([10., 0.])
px = np.array([.5, .5])
rate = information_rate(rate_map, px)
spec = information_specificity(rate_map, px)
assert rate == 5 # bits/s
assert spec == 1 # bits/spike
rate_map = np.array([
[10., 0.],
[10., 0.],
])
px = np.array([
[.25, .25],
[.25, .25],
])
rate = information_rate(rate_map, px)
spec = information_specificity(rate_map, px)
assert rate == 5 # bits/s
assert spec == 1 # bits/spike
# Spikes in 1/4 of the arena, uniform occupancy
rate_map = np.array([10., 0., 0., 0.])
px = np.array([.25, .25, .25, .25])
rate = information_rate(rate_map, px)
spec = information_specificity(rate_map, px)
assert rate == 5 # bits/s
assert spec == 2 # bits/spike
rate_map = np.array([
[10., 0.],
[ 0., 0.],
])
px = np.array([
[.25, .25],
[.25, .25],
])
rate = information_rate(rate_map, px)
spec = information_specificity(rate_map, px)
assert rate == 5 # bits/s
assert spec == 2 # bits/spike
