Coverage for lasso/dimred/svd/test_plot_betas_clusters.py: 100%

1import hashlib 

2from unittest import TestCase 

3 

4import numpy as np 

5from lasso.dimred.svd.plot_beta_clusters import plot_clusters_js 

6 

7 

8def hash_str(data: str) -> str: 

9 """hashes a string""" 

10 

11 hasher1 = hashlib.sha256() 

12 hasher1.update(data.encode("utf-8")) 

13 return hasher1.hexdigest() 

14 

15 

16class TestBetaViz(TestCase): 

17 def test_plot_clusters_js(self): 

18 """Veryfies correct output .html file""" 

19 

20 betas = [np.array([[1, 1, 1], [1, 2, 3]])] 

21 ids = np.array([["sample0", "sample1"]]) 

22 

23 html_page_str = plot_clusters_js( 

24 betas, ids, "", mark_timestamp=False, filename="testpage", write=False, show_res=False 

25 ) 

26 

27 self.assertIsInstance(html_page_str, str) 

28 if isinstance(html_page_str, str): 

29 html_page_hash = hash_str(html_page_str) 

30 

31 desired_hash = "0af9f14ded18d4ac479ac2a07d46eb773ae93a61245e57e650115cc51522c54f" 

32 self.assertEqual(html_page_hash, desired_hash)