def test_operators(): sig = Signal(np.array([0.0]), name="sig") assert fnmatch(repr(TimeUpdate(sig, sig)), "<TimeUpdate at 0x*>") assert fnmatch(repr(TimeUpdate(sig, sig, tag="tag")), "<TimeUpdate 'tag' at 0x*>") assert fnmatch(repr(Reset(sig)), "<Reset at 0x*>") assert fnmatch(repr(Reset(sig, tag="tag")), "<Reset 'tag' at 0x*>") assert fnmatch(repr(Copy(sig, sig)), "<Copy at 0x*>") assert fnmatch(repr(Copy(sig, sig, tag="tag")), "<Copy 'tag' at 0x*>") assert fnmatch(repr(ElementwiseInc(sig, sig, sig)), "<ElementwiseInc at 0x*>") assert fnmatch(repr(ElementwiseInc(sig, sig, sig, tag="tag")), "<ElementwiseInc 'tag' at 0x*>") assert fnmatch(repr(DotInc(sig, sig, sig)), "<DotInc at 0x*>") assert fnmatch(repr(DotInc(sig, sig, sig, tag="tag")), "<DotInc 'tag' at 0x*>") assert fnmatch(repr(SimPyFunc(sig, lambda x: 0.0, True, sig)), "<SimPyFunc at 0x*>") assert fnmatch( repr(SimPyFunc(sig, lambda x: 0.0, True, sig, tag="tag")), "<SimPyFunc 'tag' at 0x*>", ) assert fnmatch(repr(SimPES(sig, sig, sig, 0.1)), "<SimPES at 0x*>") assert fnmatch(repr(SimPES(sig, sig, sig, 0.1, tag="tag")), "<SimPES 'tag' at 0x*>") assert fnmatch(repr(SimBCM(sig, sig, sig, sig, 0.1)), "<SimBCM at 0x*>") assert fnmatch(repr(SimBCM(sig, sig, sig, sig, 0.1, tag="tag")), "<SimBCM 'tag' at 0x*>") assert fnmatch(repr(SimOja(sig, sig, sig, sig, 0.1, 1.0)), "<SimOja at 0x*>") assert fnmatch(repr(SimOja(sig, sig, sig, sig, 0.1, 1.0, tag="tag")), "<SimOja 'tag' at 0x*>") assert fnmatch(repr(SimVoja(sig, sig, sig, sig, 1.0, sig, 1.0)), "<SimVoja at 0x*>") assert fnmatch( repr(SimVoja(sig, sig, sig, sig, 0.1, sig, 1.0, tag="tag")), "<SimVoja 'tag' at 0x*>", ) assert fnmatch(repr(SimRLS(sig, sig, sig, sig)), "<SimRLS at 0x*>") assert fnmatch( repr(SimRLS(sig, sig, sig, sig, tag="tag")), "<SimRLS 'tag' at 0x*>", ) assert fnmatch(repr(SimNeurons(LIF(), sig, {"sig": sig})), "<SimNeurons at 0x*>") assert fnmatch( repr(SimNeurons(LIF(), sig, {"sig": sig}, tag="tag")), "<SimNeurons 'tag' at 0x*>", ) assert fnmatch(repr(SimProcess(WhiteNoise(), sig, sig, sig)), "<SimProcess at 0x*>") assert fnmatch( repr(SimProcess(WhiteNoise(), sig, sig, sig, tag="tag")), "<SimProcess 'tag' at 0x*>", )
def test_neuron_types(): check_init_args(Direct, ["initial_state"]) check_repr(Direct()) assert repr(Direct()) == "Direct()" check_init_args(RectifiedLinear, ["amplitude", "initial_state"]) check_repr(RectifiedLinear()) check_repr(RectifiedLinear(amplitude=2)) assert repr(RectifiedLinear()) == "RectifiedLinear()" check_init_args(SpikingRectifiedLinear, ["amplitude", "initial_state"]) check_repr(SpikingRectifiedLinear()) check_repr(SpikingRectifiedLinear(amplitude=2)) check_repr( SpikingRectifiedLinear(initial_state={"voltage": Choice([1.0])})) assert repr(SpikingRectifiedLinear()) == "SpikingRectifiedLinear()" check_init_args(Sigmoid, ["tau_ref", "initial_state"]) check_repr(Sigmoid()) check_repr(Sigmoid(tau_ref=0.1)) assert repr(Sigmoid()), "Sigmoid()" assert repr(Sigmoid(tau_ref=0.001)) == "Sigmoid(tau_ref=0.001)" check_init_args(LIFRate, ["tau_rc", "tau_ref", "amplitude", "initial_state"]) check_repr(LIFRate()) check_repr(LIFRate(tau_rc=0.1)) check_repr(LIFRate(tau_ref=0.1)) check_repr(LIFRate(amplitude=2)) check_repr(LIFRate(tau_rc=0.05, tau_ref=0.02)) check_repr(LIFRate(tau_rc=0.05, amplitude=2)) check_repr(LIFRate(tau_ref=0.02, amplitude=2)) check_repr(LIFRate(tau_rc=0.05, tau_ref=0.02, amplitude=2)) assert repr(LIFRate()) == "LIFRate()" assert repr(LIFRate(tau_rc=0.01, tau_ref=0)) == "LIFRate(tau_rc=0.01, tau_ref=0)" check_init_args( LIF, ["tau_rc", "tau_ref", "min_voltage", "amplitude", "initial_state"]) check_repr(LIF()) check_repr(LIF(tau_rc=0.1)) check_repr(LIF(tau_ref=0.1)) check_repr(LIF(amplitude=2)) check_repr(LIF(min_voltage=-0.5)) check_repr(LIF(tau_rc=0.05, tau_ref=0.02)) check_repr(LIF(tau_rc=0.05, amplitude=2)) check_repr(LIF(tau_ref=0.02, amplitude=2)) check_repr(LIF(tau_rc=0.05, tau_ref=0.02, amplitude=2)) check_repr(LIF(tau_rc=0.05, tau_ref=0.02, min_voltage=-0.5, amplitude=2)) check_repr(LIF(initial_state={"refractory_time": Choice([0.1])})) assert repr(LIF()) == "LIF()" assert repr(LIF(tau_rc=0.01, tau_ref=0)) == "LIF(tau_rc=0.01, tau_ref=0)" check_init_args( AdaptiveLIFRate, ["tau_n", "inc_n", "tau_rc", "tau_ref", "amplitude", "initial_state"], ) check_repr(AdaptiveLIFRate()) check_repr(AdaptiveLIFRate(tau_n=0.1)) check_repr(AdaptiveLIFRate(inc_n=0.5)) check_repr(AdaptiveLIFRate(tau_rc=0.1)) check_repr(AdaptiveLIFRate(tau_ref=0.1)) check_repr(AdaptiveLIFRate(amplitude=2)) check_repr( AdaptiveLIFRate(tau_n=0.1, inc_n=0.5, tau_rc=0.05, tau_ref=0.02, amplitude=2)) check_repr(AdaptiveLIFRate(initial_state={"adaptation": Choice([0.1])})) assert repr(AdaptiveLIFRate()) == "AdaptiveLIFRate()" assert (repr(AdaptiveLIFRate( tau_rc=0.01, tau_n=0.5, inc_n=0.02)) == "AdaptiveLIFRate(tau_n=0.5, inc_n=0.02, tau_rc=0.01)") check_init_args( AdaptiveLIF, [ "tau_n", "inc_n", "tau_rc", "tau_ref", "min_voltage", "amplitude", "initial_state", ], ) check_repr(AdaptiveLIF()) check_repr(AdaptiveLIF(tau_n=0.1)) check_repr(AdaptiveLIF(inc_n=0.5)) check_repr(AdaptiveLIF(tau_rc=0.1)) check_repr(AdaptiveLIF(tau_ref=0.1)) check_repr(AdaptiveLIF(min_voltage=-0.5)) check_repr( AdaptiveLIF( tau_n=0.1, inc_n=0.5, tau_rc=0.05, tau_ref=0.02, min_voltage=-0.5, amplitude=2, )) check_repr(AdaptiveLIF(initial_state={"adaptation": Choice([0.1])})) assert repr(AdaptiveLIF()) == "AdaptiveLIF()" assert (repr(AdaptiveLIF( tau_rc=0.01, tau_n=0.5, inc_n=0.02)) == "AdaptiveLIF(tau_n=0.5, inc_n=0.02, tau_rc=0.01)") check_init_args( Izhikevich, [ "tau_recovery", "coupling", "reset_voltage", "reset_recovery", "initial_state", ], ) check_repr(Izhikevich()) check_repr(Izhikevich(tau_recovery=0.1)) check_repr(Izhikevich(coupling=0.3)) check_repr(Izhikevich(reset_voltage=-1)) check_repr(Izhikevich(reset_recovery=5)) check_repr( Izhikevich(tau_recovery=0.1, coupling=0.3, reset_voltage=-1, reset_recovery=5)) check_repr(Izhikevich(initial_state={"recovery": Choice([0.1])})) assert repr(Izhikevich()) == "Izhikevich()" assert (repr( Izhikevich(tau_recovery=0.01, coupling=0.5, reset_voltage=-60, reset_recovery=6)) == "Izhikevich(tau_recovery=0.01, coupling=0.5, reset_voltage=-60, " "reset_recovery=6)") check_init_args(Tanh, ["tau_ref", "initial_state"]) check_repr(Tanh()) check_repr(Tanh(tau_ref=0.1))