Resurrect cost_size (w/ DeprecationWarning), write test for it.

Author: dabliss

brainx/tests/test_util.py

@@ -16,6 +16,12 @@
 #-----------------------------------------------------------------------------
 # Functions
 #-----------------------------------------------------------------------------
+
+def test_cost_size():
+    n_nodes = 5
+    npt.assert_warns(DeprecationWarning, util.cost_size, n_nodes)
+    
+
 def test_apply_cost():
     corr_mat = np.array([[0.0, 0.5, 0.3, 0.2, 0.1],
                          [0.5, 0.0, 0.4, 0.1, 0.2],

brainx/util.py

@@ -6,8 +6,9 @@
 #-----------------------------------------------------------------------------
 from __future__ import print_function
 
-import numpy as np
+import warnings
 
+import numpy as np
 import networkx as nx
 
 #-----------------------------------------------------------------------------
@@ -60,6 +61,14 @@ def format_matrix2(data,s,sc,c,lk,co,idc = [],costlist=[],nouptri = False):
     return cmat
 
 
+def cost_size(nnodes):
+    warnings.warn('deprecated: use make_cost_array', DeprecationWarning)
+    tot_edges = 0.5 * nnodes * (nnodes - 1)
+    costs = np.array(range(int(tot_edges) + 1), dtype=float) / tot_edges
+    edges_short = tot_edges / 2
+    return costs, tot_edges, edges_short
+
+
 def make_cost_array(n_nodes, cost=0.5):
     """Make cost array of length cost * (the number of possible edges).