Use SeleniumBase's virtual_display package

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -3256,7 +3256,8 @@ def setUp(self, masterqa_mode=False):
                 self.case_start_time = int(time.time() * 1000)
             if self.headless:
                 try:
-                    from pyvirtualdisplay import Display
+                    # from pyvirtualdisplay import Display  # Skip for own lib
+                    from seleniumbase.virtual_display.display import Display
                     self.display = Display(visible=0, size=(1440, 1880))
                     self.display.start()
                     self.headless_active = True

seleniumbase/plugins/selenium_plugin.py

@@ -285,7 +285,8 @@ def beforeTest(self, test):
             test.test.headed = True
         if self.options.headless:
             try:
-                from pyvirtualdisplay import Display
+                # from pyvirtualdisplay import Display  # Skip for own lib
+                from seleniumbase.virtual_display.display import Display
                 self.display = Display(visible=0, size=(1440, 1880))
                 self.display.start()
                 self.headless_active = True