Use chain.from_iterable in mocking.py

miki725 · Jun 17, 2020 · 76f7833 · 76f7833
1 parent f8af888
commit 76f7833
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/alchemy_mock/mocking.py b/alchemy_mock/mocking.py
@@ -434,13 +434,8 @@ def _get_data(self, *args, **kwargs):
             if _mock_name == "get":
                 query_call = [c for c in previous_calls if c[0] == "query"][0]
                 results = list(
-                    chain(
-                        *[
-                            result
-                            for calls, result in sorted_mock_data
-                            if query_call in calls
-                        ]
-                    )
+                    chain.from_iterable(result for calls, result in sorted_mock_data
+                                        if query_call in calls)
                 )
                 return self.boundary[_mock_name](results, *args, **kwargs)