chore: import upstream snapshot with attribution

test/legacy_test/test_imperative_data_loader_exit_func.py

@@ -0,0 +1,82 @@
+# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import multiprocessing
+import queue
+import signal
+import time
+import unittest
+
+from paddle.base.reader import (
+    CleanupFuncRegistrar,
+    _cleanup,
+    multiprocess_queue_set,
+)
+
+# NOTE: These special functions cannot be detected by the existing coverage mechanism,
+# so the following unittests are added for these internal functions.
+
+
+class TestDygraphDataLoaderCleanUpFunc(unittest.TestCase):
+    def setUp(self):
+        self.capacity = 10
+
+    def test_clear_queue_set(self):
+        test_queue = queue.Queue(self.capacity)
+        multiprocess_queue_set.add(test_queue)
+        for i in range(0, self.capacity):
+            test_queue.put(i)
+        _cleanup()
+
+
+class TestRegisterExitFunc(unittest.TestCase):
+    # This function does not need to be implemented in this case
+    def none_func(self):
+        pass
+
+    def test_not_callable_func(self):
+        exception = None
+        try:
+            CleanupFuncRegistrar.register(5)
+        except TypeError as ex:
+            self.assertIn("is not callable", str(ex))
+            exception = ex
+        self.assertIsNotNone(exception)
+
+    def test_old_handler_for_sigint(self):
+        CleanupFuncRegistrar.register(
+            function=self.none_func, signals=[signal.SIGINT]
+        )
+
+    def test_signal_wrapper_by_sigchld(self):
+        # This function does not need to be implemented in this case
+        def __test_process__():
+            pass
+
+        CleanupFuncRegistrar.register(
+            function=self.none_func, signals=[signal.SIGCHLD]
+        )
+
+        exception = None
+        try:
+            test_process = multiprocessing.Process(target=__test_process__)
+            test_process.start()
+            time.sleep(3)
+        except SystemExit as ex:
+            exception = ex
+        self.assertIsNotNone(exception)
+
+
+if __name__ == '__main__':
+    unittest.main()