added simple test data

TuftsBCB · Dec 20, 2016 · a507dbf · a507dbf
1 parent aefa89f
commit a507dbf
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/testdata/test_network.ppi b/testdata/test_network.ppi
@@ -0,0 +1,6 @@
+0	1	.25
+0	2	.25
+1	2	.25
+2	3	.80
+3	4	.85
+1	4	.50
diff --git a/testdata/test_seed.txt b/testdata/test_seed.txt
@@ -0,0 +1,2 @@
+0
+1
diff --git a/walker.py b/walker.py
@@ -192,15 +192,11 @@ def _build_og(self, original_ppi):
         # parse network input
         for line in graph_fp.readlines():
             split_line = line.rstrip().split('\t')
-            if len(split_line) > 2:
+            if len(split_line) > 3:
                 # assume input graph is in the form of HIPPIE network
-                try:
-                    edge_list.append((split_line[1], split_line[3],
-                                      float(split_line[4])))
-                except ValueError:
-                    print line
-                    exit()
-            elif len(split_line) < 2:
+                edge_list.append((split_line[1], split_line[3],
+                                  float(split_line[4])))
+            elif len(split_line) < 3:
                 # assume input graph is a simple edgelist without weights
                 edge_list.append((split_line[0], split_line[1], float(1)))
             else: