Changeset 437 for trunk/test/test_stgen.py

Timestamp:

10/01/09 17:29:53 (3 years ago)

Author:

emuller

Message:

Doc string for inh_gamma_generator, and new test

Files:

• trunk/test/test_stgen.py (modified) (3 diffs)

trunk/test/test_stgen.py

@@ -187 +187 @@
 
 
+    def testInhGammaBasic(self):
+
+        from numpy import array
+        import NeuroTools.signals
+
+        stg = stgen.StGen()
+
+        st = stg.inh_gamma_generator(array([3.0,3.0]),array([1.0/100.0/3.0,1.0/200.0/3.0]),array([500.0,1500.0]),2500.0,array=True)
+
+        assert(type(st)==numpy.ndarray)
+
+        st = stg.inh_gamma_generator(array([3.0,3.0]),array([1.0/100.0/3.0,1.0/200.0/3.0]),array([500.0,1500.0]),2500.0,array=False)
+
+        assert(type(st)==NeuroTools.signals.spikes.SpikeTrain)
+
+
+
     def testStatsInhGamma(self):
 
@@ -192 +209 @@
 
         from numpy import array
+        import NeuroTools.signals
 
         stg = stgen.StGen()
@@ -198 +216 @@
 
         st = stg.inh_gamma_generator(array([3.0,3.0]),array([1.0/100.0/3.0,1.0/200.0/3.0]),array([500.0,1500.0]),2500.0,array=True)
+
+        assert(type(st)==numpy.ndarray)