added more tests for ensembling

ntalluri · Aug 15, 2024 · 8c71ad8 · 8c71ad8
1 parent b6610ae
commit 8c71ad8
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/test/ml/expected/expected-ensemble-network-empty.tsv b/test/ml/expected/expected-ensemble-network-empty.tsv
@@ -0,0 +1 @@
+Node1	Node2	Frequency	Direction
diff --git a/test/ml/expected/expected-ensemble-network-single.tsv b/test/ml/expected/expected-ensemble-network-single.tsv
@@ -0,0 +1,2 @@
+Node1	Node2	Frequency	Direction
+L	M	1.0	U
diff --git a/test/ml/test_ml.py b/test/ml/test_ml.py
@@ -95,3 +95,25 @@ def test_ensemble_network(self):
         expected = expected.round(5)
 
         assert en.equals(expected)
+
+    def test_ensemble_network_single_line(self):
+        dataframe = ml.summarize_networks([INPUT_DIR + 'test-data-single/single.txt'])
+        ml.ensemble_network(dataframe, OUT_DIR + 'ensemble-network-single.tsv')
+
+        en = pd.read_table(OUT_DIR + 'ensemble-network-single.tsv')
+        en = en.round(5)
+        expected = pd.read_table(EXPECT_DIR + 'expected-ensemble-network-single.tsv')
+        expected = expected.round(5)
+
+        assert en.equals(expected)
+
+    def test_ensemble_network_empty(self):
+        dataframe = ml.summarize_networks([INPUT_DIR + 'test-data-empty/empty.txt'])
+        ml.ensemble_network(dataframe, OUT_DIR + 'ensemble-network-empty.tsv')
+
+        en = pd.read_table(OUT_DIR + 'ensemble-network-empty.tsv')
+        en = en.round(5)
+        expected = pd.read_table(EXPECT_DIR + 'expected-ensemble-network-empty.tsv')
+        expected = expected.round(5)
+
+        assert en.equals(expected)