def test_build_record(): simulator = Mock() simulator.state.write_on_end = [] record_function = common.build_record(simulator) assert isfunction(record_function) source = BasePopulation() source.record = Mock() record_function(('v', 'spikes'), source, "filename") source.record.assert_called_with(('v', 'spikes'), to_file="filename", sampling_interval=None)
def test_build_record(): simulator = Mock() simulator.state.write_on_end = [] record_function = common.build_record(simulator) assert isfunction(record_function) source = BasePopulation() source.record = Mock() record_function(("v", "spikes"), source, "filename") source.record.assert_called_with(("v", "spikes"), to_file="filename") assert_equal(simulator.state.write_on_end, [(source, ("v", "spikes"), "filename")])
def test_build_record(): simulator = Mock() simulator.recorder_list = [] record_function = common.build_record("foo", simulator) assert isfunction(record_function) source = BasePopulation() source._record = Mock() source.recorders = {'foo': Mock()} record_function(source, "filename") source._record.assert_called_with("foo", to_file="filename") assert_equal(simulator.recorder_list, [source.recorders['foo']])
def test_build_record_with_assembly(): simulator = Mock() simulator.state.write_on_end = [] record_function = common.build_record(simulator) assert isfunction(record_function) p1 = BasePopulation() p2 = BasePopulation() common.Assembly._simulator = None source = common.Assembly(p1, p2) source.record = Mock() record_function('foo', source, "filename") source.record.assert_called_with('foo', to_file="filename", sampling_interval=None)
def test_build_record_with_assembly(): simulator = Mock() simulator.state.write_on_end = [] record_function = common.build_record(simulator) assert isfunction(record_function) p1 = BasePopulation() p2 = BasePopulation() source = common.Assembly(p1, p2) source.record = Mock() record_function('foo', source, "filename") source.record.assert_called_with('foo', to_file="filename") assert_equal(simulator.state.write_on_end, [ (source, 'foo', "filename") ]) # not sure this is what we want - won't file get over-written?
def test_set(): cells = BasePopulation() cells.set = Mock() common.set(cells, param="val") cells.set.assert_called_with(param="val")