Update tests for new plots

brian-team · May 3, 2016 · e1f5c99 · e1f5c99
1 parent 4608438
commit e1f5c99
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/brian2tools/tests/test_plotting.py b/brian2tools/tests/test_plotting.py
@@ -3,6 +3,7 @@
 '''
 import matplotlib
 matplotlib.use('Agg')
+
 from brian2 import *
 from brian2tools import *
 
@@ -50,12 +51,29 @@ def test_plot_synapses():
     close()
     plot_synapses(synapses.i, synapses.j)
     close()
+    plot_synapses(synapses.i, synapses.j, plot_type='scatter')
+    close()
+    plot_synapses(synapses.i, synapses.j, plot_type='image')
+    close()
+    plot_synapses(synapses.i, synapses.j, plot_type='hexbin')
+    close()
     plot_synapses(synapses.i, synapses.j, synapses.w)
     close()
+    plot_synapses(synapses.i, synapses.j, synapses.w, plot_type='scatter')
+    close()
+    plot_synapses(synapses.i, synapses.j, synapses.w, plot_type='image')
+    close()
+    plot_synapses(synapses.i, synapses.j, synapses.w, plot_type='hexbin')
+    close()
 
     synapses.connect('i > 5')  # More than one synapse per connection
     brian_plot(synapses)
     close()
+    # It should be possible to plot synaptic variables for multiple connections
+    # with hexbin
+    plot_synapses(synapses.i, synapses.j, synapses.w, plot_type='hexbin')
+    close()
+
 
 def test_plot_morphology():
     # Only testing 2D plotting for now