Added test method for Sink.output

test_pdbuddy/__init__.py

@@ -101,6 +101,18 @@ class SinkTestCase(unittest.TestCase):
             # essentially test_get_cfg_index.
             self.assertIsInstance(pdbs.get_cfg(0), pdbuddy.SinkConfig)
 
+    def test_output(self):
+        try:
+            self.pdbs.output(False)
+            self.assertFalse(self.pdbs.output())
+
+            self.pdbs.output(True)
+            self.assertTrue(self.pdbs.output())
+        except KeyError:
+            self.skipTest("Command output not supported")
+        except ValueError:
+            self.skipTest("Unknown value returned by PD Buddy Sink")
+
 
 class SinkConfigTestCase(unittest.TestCase):