support for if __name__ == __main__

(cherry picked from commit e7edf5b)
AirtestProject · Mar 22, 2023 · aeda723 · aeda723
1 parent e17a441
commit aeda723
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/airtest/cli/runner.py b/airtest/cli/runner.py
@@ -38,6 +38,8 @@ def setUp(self):
                     dev.start_recording()
                 except:
                     traceback.print_exc()
+        # support for if __name__ == '__main__'
+        self.scope['__name__'] = '__main__'
 
     def tearDown(self):
         if self.args.log and self.args.recording: