Coverage for common/future.py: 36%

1""" 

2crate_anon/preprocess/autoimport_db.py 

3 

4=============================================================================== 

5 

6 Copyright (C) 2015, University of Cambridge, Department of Psychiatry. 

7 Created by Rudolf Cardinal (rnc1001@cam.ac.uk). 

8 

9 This file is part of CRATE. 

10 

11 CRATE is free software: you can redistribute it and/or modify 

12 it under the terms of the GNU General Public License as published by 

13 the Free Software Foundation, either version 3 of the License, or 

14 (at your option) any later version. 

15 

16 CRATE is distributed in the hope that it will be useful, 

17 but WITHOUT ANY WARRANTY; without even the implied warranty of 

18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

19 GNU General Public License for more details. 

20 

21 You should have received a copy of the GNU General Public License 

22 along with CRATE. If not, see <https://www.gnu.org/licenses/>. 

23 

24=============================================================================== 

25 

26**Python code that will be in the Python standard library in higher versions of 

27Python.** 

28 

29""" 

30 

31from itertools import islice 

32from typing import Any, Generator, Iterable, List 

33import warnings 

34 

35 

36def batched( 

37 iterable: Iterable[Any], n: int 

38) -> Generator[List[Any], None, None]: 

39 """ 

40 Batch data into lists of length n. The last batch may be shorter. 

41 

42 batched('ABCDEFG', 3) --> ABC DEF G 

43 

44 From Python 3.12, this is itertools.batched(). See 

45 

46 - https://stackoverflow.com/questions/8290397 

47 - https://docs.python.org/3/library/itertools.html#itertools.batched 

48 """ 

49 warnings.warn( 

50 "When Python 3.12 is the minimum for CRATE, use itertools.batched " 

51 "instead of crate_anon.common.future.batched", 

52 FutureWarning, 

53 ) 

54 it = iter(iterable) 

55 while True: 

56 batch = list(islice(it, n)) 

57 if not batch: 

58 return 

59 yield batch