fix: function testS

microsoft · Jun 6, 2017 · ff8a316 · ff8a316
1 parent 347c0b2
commit ff8a316
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/qcodes/tests/test_channels.py b/qcodes/tests/test_channels.py
@@ -2,6 +2,7 @@
 import unittest
 
 from qcodes.tests.instrument_mocks import DummyChannelInstrument, DummyChannel
+from qcodes.instrument.channel import ChannelList
 from qcodes.utils.validators import Numbers
 from qcodes.instrument.parameter import ManualParameter
 
@@ -125,8 +126,17 @@ def test_combine_channels(self, setpoints):
         self.assertEquals(self.instrument.channels.temperature(), expected)
 
     def test_channel_parameters(self):
-            self.assertTrue('temperature' in self.instrument.channels.parameters)
-            self.assertEqual(len(self.instrument.channels.parameters), 1)
+        self.assertTrue('temperature' in self.instrument.channels.parameters)
+        self.assertEqual(len(self.instrument.channels.parameters), 1)
+
+
+    def test_channel_functions(self):
+        dc = DummyChannel(self.instrument, 'Chan' + 'bar', 'bar')
+        dc.add_function('dummyfunc', call_cmd='foobar')
+        self.assertTrue('dummyfunc' in dc.functions)
+        dcl = ChannelList(self.instrument, 'DummyChannelList', DummyChannel)
+        dcl.append(dc)
+        self.assertTrue('dummyfunc' in dcl.functions)
 
 class TestChannelsLoop(TestCase):