36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import unittest
|
|
import sys
|
|
sys.path.append('../..')
|
|
from system.dataset import *
|
|
from scipy import rand,shape
|
|
|
|
class DatasetTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
dim_0_ids = ('sample_a','sample_b')
|
|
dim_1_ids = ('gene_a','gene_b','gene_c')
|
|
dim_labels = ('samples','genes')
|
|
identifiers= [(dim_labels[0],dim_0_ids),(dim_labels[1],dim_1_ids)]
|
|
self.array = rand(2,3)
|
|
self.testdata = Dataset(self.array,identifiers)
|
|
|
|
def testCreation(self):
|
|
data = self.testdata
|
|
assert data._array == self.array
|
|
assert 'sample_a' in data['samples'].keys()
|
|
assert data['samples']['sample_b']==1
|
|
assert 'gene_c' in data['genes'].keys()
|
|
assert data['genes']['gene_c']==2
|
|
|
|
#def testExtraction(self):
|
|
# ids = ['gene_a','gene_b']
|
|
# dim_name = 'genes'
|
|
# subset = self.testdata.extract_data(ids,dim_name)
|
|
# assert shape(subset._data) == (2,2)
|
|
# assert subset.ids[dim_name].keys() == ids
|
|
# assert subset.ids[dim_name].values() == [0,1]
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|