def drop_test(): import numpy as np import pyemu arr = np.arange(0,12) arr.resize(4,3) first = pyemu.Jco(x=arr,col_names=["p2","p1","p3"],row_names=["o4","o1","o3","o2"]) print(first) first.drop(["o2","o1"],axis=0) print(first) first.drop(["p1","p3"],axis=1) print(first) assert first.row_names == ["o4","o3"] assert first.col_names == ["p2"] t_array = np.atleast_2d(np.array([0,6])).transpose() print(first.x,t_array) assert np.array_equal(first.x,t_array)
def get_test(): import numpy as np import pyemu arr = np.arange(0,12) arr.resize(4,3) first = pyemu.Jco(x=arr,col_names=["p2","p1","p3"],row_names=["o4","o1","o3","o2"]) #print(first) #second = first.get(row_names=["o1","o3"]) second = first.get(row_names=first.row_names) assert np.array_equal(first.x,second.x) cov1 = pyemu.Cov(x=np.atleast_2d(np.arange(10)).transpose(),names=["c{0}".format(i) for i in range(10)],isdiagonal=True) print(cov1) cov2 = cov1.get(row_names=cov1.row_names) print(cov2)