Coverage for tests / tests_registers / test_type_register.py: 100%

1# SPDX-FileCopyrightText: Copyright INRIA 

2# 

3# SPDX-License-Identifier: LGPL-3.0-only 

4# 

5# Copyright INRIA 

6# 

7# This file is part of PhysioBlocks, a library mostly developed by the 

8# [Ananke project-team](https://team.inria.fr/ananke) at INRIA. 

9# 

10# Authors: 

11# - Colin Drieu 

12# - Dominique Chapelle 

13# - François Kimmig 

14# - Philippe Moireau 

15# 

16# PhysioBlocks is free software: you can redistribute it and/or modify it under the 

17# terms of the GNU Lesser General Public License as published by the Free Software 

18# Foundation, version 3 of the License. 

19# 

20# PhysioBlocks is distributed in the hope that it will be useful, but WITHOUT ANY 

21# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A 

22# PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. 

23# 

24# You should have received a copy of the GNU Lesser General Public License along with 

25# PhysioBlocks. If not, see <https://www.gnu.org/licenses/>. 

26 

27from physioblocks.registers.type_register import ( 

28 get_registered_type, 

29 get_registered_type_id, 

30 is_registered, 

31 register_type, 

32) 

33 

34CLASS_A_ID = "class_a" 

35CLASS_B_ID = "class_b" 

36 

37 

38def test_type_register(): 

39 @register_type(CLASS_A_ID) 

40 class ClassA: 

41 pass 

42 

43 assert is_registered(CLASS_A_ID) is True 

44 assert is_registered(ClassA) is True 

45 assert get_registered_type(CLASS_A_ID) == ClassA 

46 assert get_registered_type_id(ClassA) == CLASS_A_ID