Coverage for tests / tests_library / tests_model_components / test_rheology.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 

27import numpy as np 

28import pytest 

29 

30from physioblocks.computing.quantities import Quantity 

31from physioblocks.library.model_components.rheology import ( 

32 RheologyFiberAdditiveModelComponent, 

33) 

34from physioblocks.simulation.state import State 

35from physioblocks.simulation.time_manager import Time 

36from physioblocks.utils.gradient_test_utils import gradient_test_from_model 

37 

38 

39@pytest.fixture 

40def ref_model() -> RheologyFiberAdditiveModelComponent: 

41 return RheologyFiberAdditiveModelComponent( 

42 disp=Quantity(0.15), 

43 fib_deform=Quantity(0.1), 

44 active_tension_discr=Quantity(2000.0), 

45 radius=Quantity(0.03), 

46 series_stiffness=Quantity(100000), 

47 damping_parallel=Quantity(70.0), 

48 time=Time(0.0), 

49 ) 

50 

51 

52class TestRheologyFiberAdditiveModelComponent: 

53 def test_check_gradient(self, ref_model: RheologyFiberAdditiveModelComponent): 

54 state = State() 

55 state["fib_deform"] = ref_model.fib_deform 

56 state["disp"] = ref_model.disp 

57 state["active_tension_discr"] = ref_model.active_tension_discr 

58 

59 ref_model.time.update(0.001) 

60 magnitudes = np.array([0.14, 0.015, 1999.3]) 

61 

62 assert gradient_test_from_model(ref_model, state, magnitudes)