Skip to content

Commit efeb6bb

Browse files
committed
Removed all stochasticity from the dm test
1 parent 566491b commit efeb6bb

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed
0 Bytes
Binary file not shown.
3.39 KB
Binary file not shown.
-13.6 KB
Binary file not shown.

brainbox/tests/test_modeling.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,17 +11,17 @@ class TestModeling(unittest.TestCase):
1111
def setUp(self):
1212
# Params for test
1313
self.binwidth = 0.02
14-
self.rng = np.random.default_rng(seed=112358)
1514

1615
# Generate fake trial start, stimon, feedback, and end times
1716
starts = np.array([0, 1.48, 2.93, 4.67, 6.01, 7.31, 8.68, 9.99, 11.43, 12.86])
1817
ends = np.array([1.35, 2.09, 3.53, 5.23, 6.58, 7.95, 9.37, 11.31, 12.14, 13.26])
1918
stons = starts + 0.1
20-
fdbks = stons + self.rng.normal(loc=0.1, scale=0.05, size=10)
19+
fdbks = np.array([0.24, 1.64, 3.15, 4.81, 6.23, 7.50, 8.91, 10.16, 11.64, 13.05])
2120

2221
# Figure out how many bins each trial is and generate non-monotonic trace of fake wheel
23-
trlens = self.binf(ends - starts)
24-
fakewheels = [np.cumsum(self.rng.normal(loc=0.01, scale=0.01, size=(x))) for x in trlens]
22+
whlpath = Path(__file__).parent.joinpath('fixtures', 'design_wheel_traces_test.p')
23+
if whlpath.exists():
24+
fakewheels = np.load(whlpath, allow_pickle=True)
2525

2626
# Store trialsdf for later use
2727
self.trialsdf = pd.DataFrame({'trial_start': starts,

0 commit comments

Comments
 (0)