ENH: add video test vcp's to chooser

kcjengr · Mar 24, 2019 · f19b374 · f19b374
1 parent 91b8b6d
commit f19b374
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/qtpyvcp/vcp_chooser/vcp_chooser.py b/qtpyvcp/vcp_chooser/vcp_chooser.py
@@ -36,6 +36,17 @@ def __init__(self, opts):
             child = QTreeWidgetItem(category)
             child.setText(0, self.get_vcp_data(entry_point))
 
+        # test VCP section
+        category = QTreeWidgetItem(self.vcpTreeView)
+        category.setText(0, 'Video Test VCPs')
+        category.setFlags(Qt.ItemIsEnabled)
+
+        # add example VCPs to the treeview
+        for entry_point in iter_entry_points(group='qtpyvcp.test_vcp'):
+            child = QTreeWidgetItem(category)
+            child.setText(0, self.get_vcp_data(entry_point))
+
+
         # installed VCP section
         category = QTreeWidgetItem(self.vcpTreeView)
         category.setText(0, 'Installed VCPs')

diff --git a/setup.py b/setup.py
@@ -113,5 +113,10 @@ def data_files_from_dirs(data_dirs):
             'probe_basic_vertical=examples.probe_basic_vertical',
             'probe_basic_lathe=examples.probe_basic_lathe',
         ],
+        'qtpyvcp.test_vcp': [
+            'vtk_test=video_tests.vkt_test',
+            'opengl_test=video_tests.opengl_test',
+            'qtpyvcp_test=video_tests.qtpyvcp_test',
+        ],
     },
 )