Coverage for tests\utils\test_frequency.py: 100.00%

1# -*- coding: utf-8 -*- 

2 

3""" 

4* Name: interactive-clustering/tests/utils/test_frequency.py 

5* Description: Unittests for the `utils.frequency` module. 

6* Author: Erwan SCHILD 

7* Created: 17/03/2021 

8* Licence: CeCILL (https://cecill.info/licences.fr.html) 

9""" 

10 

11# ============================================================================== 

12# IMPORT PYTHON DEPENDENCIES 

13# ============================================================================== 

14 

15from cognitivefactory.interactive_clustering.utils.frequency import compute_clusters_frequency 

16 

17 

18# ============================================================================== 

19# test_compute_clusters_frequency_for_no_clustering_result 

20# ============================================================================== 

21def test_compute_clusters_frequency_for_no_clustering_result(): 

22 """ 

23 Test that the `utils.frequency.compute_clusters_frequency` works for no `clustering_result`. 

24 """ 

25 

26 # Check for no `clustering_result`. 

27 assert not compute_clusters_frequency(clustering_result={}) 

28 

29 

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

31# test_compute_clusters_frequency_for_correct_clustering_result 

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

33def test_compute_clusters_frequency_for_correct_clustering_result(): 

34 """ 

35 Test that the `utils.frequency.compute_clusters_frequency` works for correct `clustering_result`. 

36 """ 

37 

38 # Check for no `clustering_result`. 

39 assert compute_clusters_frequency( 

40 clustering_result={ 

41 "01": 0, 

42 "02": 0, 

43 "03": 0, 

44 "04": 0, 

45 "05": 0, 

46 "06": 1, 

47 "07": 1, 

48 "08": 2, 

49 "09": 2, 

50 "10": 2, 

51 } 

52 ) == { 

53 0: 0.5, 

54 1: 0.2, 

55 2: 0.3, 

56 }