Coverage for common/tests/regex_helper_tests.py: 100%

1""" 

2crate_anon/common/tests/regex_helper_tests.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 

26Unit testing. 

27 

28""" 

29 

30# ============================================================================= 

31# Imports 

32# ============================================================================= 

33 

34from unittest import TestCase 

35 

36from crate_anon.common.regex_helpers import first_n_characters_required 

37 

38 

39# ============================================================================= 

40# Unit tests 

41# ============================================================================= 

42 

43 

44class RegexHelperTests(TestCase): 

45 def test_first_n_characters_required(self) -> None: 

46 self.assertEqual(first_n_characters_required("abc", 3), "abc") 

47 self.assertEqual(first_n_characters_required("abc", 4), "abc") 

48 self.assertEqual(first_n_characters_required("abcd", 3), "abc(?:d)?") 

49 self.assertEqual( 

50 first_n_characters_required("abcde", 3), "abc(?:d(?:e)?)?" 

51 )