aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/computations/test_slink.py
diff options
context:
space:
mode:
authorMuriithi Frederick Muriuki2021-07-23 09:18:40 +0300
committerMuriithi Frederick Muriuki2021-07-23 09:18:40 +0300
commit278f1a18b59b1cadd04c50a9af35b5aece4d722d (patch)
tree7cf307f6fb0443c644fd051e51d65688c3846cf3 /tests/unit/computations/test_slink.py
parent8705e9186051fb5d11d150991c49f6f73056183b (diff)
downloadgenenetwork3-278f1a18b59b1cadd04c50a9af35b5aece4d722d.tar.gz
Add data examples for `slink`. Implement function.
Issue: https://github.com/genenetwork/gn-gemtext-threads/blob/main/topics/gn1-migration-to-gn2/clustering.gmi * gn3/computations/slink.py: Copy the function, mostly verbatim from genenetwork1. See: https://github.com/genenetwork/genenetwork1/blob/master/web/webqtl/heatmap/slink.py#L107-L138 * tests/unit/computations/test_slink.py: Add a test with some example data to test that the implementation gives the same results as that in genenetwork1
Diffstat (limited to 'tests/unit/computations/test_slink.py')
-rw-r--r--tests/unit/computations/test_slink.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/tests/unit/computations/test_slink.py b/tests/unit/computations/test_slink.py
index da17ac3..5627767 100644
--- a/tests/unit/computations/test_slink.py
+++ b/tests/unit/computations/test_slink.py
@@ -207,3 +207,14 @@ class TestSlink(TestCase):
for data in [1, "test", [], 2.945, nearest, [0]]:
with self.subTest(data=data):
self.assertEqual(slink(data), [])
+
+ def test_slink_with_data(self):
+ for data, expected in [
+ [[[0,9],[9,0]],[0,1,9]],
+ [[[0,9,3],[9,0,7],[3,7,0]],[(0,2,3),1,7]],
+ [[[0,9,3,6],[9,0,7,5],[3,7,0,9],[6,5,9,0]],[(0,2,3),(1,3,5),6]],
+ [[[0,9,3,6,11],[9,0,7,5,10],[3,7,0,9,2],[6,5,9,0,8],
+ [11,10,2,8,0]],
+ [(0,(2,4,2),3),(1,3,5),6]]]:
+ with self.subTest(data=data):
+ self.assertEqual(slink(data), expected)