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_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_set(): cells = BasePopulation() cells.set = Mock() common.set(cells, param="val") cells.set.assert_called_with(param="val")