Coverage for tests/test_fsd.py: 100%

1import typing 

2 

3from hypothesis import given, strategies as st 

4import numpy as np 

5 

6from crispatt._fsd import MonoFSD, RampFSD 

7 

8if typing.TYPE_CHECKING: 

9 from crispatt._fsd import _FSDHandler 

10 

11pancake_size_strategy = st.floats( 

12 min_value=0.1, 

13 max_value=100.0, 

14 allow_nan=False, 

15) 

16n_floes_st = st.integers(1, 20) 

17 

18 

19def mono_average(handler: MonoFSD): 

20 assert handler.average_length == handler.length 

21 

22 

23def ramp_average(handler): 

24 assert handler.average_length == (handler.length_left + handler.length_right) / 2 

25 

26 

27@given(length=pancake_size_strategy) 

28def test_mono(length: float): 

29 handler = MonoFSD(length) 

30 assert handler.length == length 

31 mono_average(handler) 

32 n_floes(handler) 

33 

34 

35@given(lengths=st.tuples(pancake_size_strategy, pancake_size_strategy)) 

36def test_ramp(lengths: tuple[float, float]): 

37 length_left, length_right = lengths 

38 handler = RampFSD(length_left, length_right) 

39 assert handler.length_left == length_left 

40 assert handler.length_right == length_right 

41 ramp_average(handler) 

42 n_floes(handler) 

43 

44 

45@given(n_floes=n_floes_st) 

46def n_floes(handler: "_FSDHandler", n_floes: int): 

47 lengths = handler.compute_lengths(n_floes) 

48 assert isinstance(lengths, np.ndarray) 

49 assert lengths.ndim == 1 

50 assert lengths.size == n_floes + 1 

51 assert np.isposinf(lengths[-1])