def test_split(self): slist = SampleList.from_folder("testdataset/cat1/", [1]) (train, test) = slist.split(0.5) self.assertEquals(len(train), 1) self.assertEquals(len(test), 1) self.assertEquals(train[0], slist[0]) self.assertEquals(test[0], slist[1])
def test_split(self): slist = SampleList.from_folder("testdataset/cat1/", [1]) (train,test) = slist.split(0.5) self.assertEquals(len(train), 1) self.assertEquals(len(test), 1) self.assertEquals(train[0], slist[0]) self.assertEquals(test[0], slist[1])
def test_toinput_response(self): slist = SampleList.from_folder("testdataset/cat1/", [1]) slist.load_data() (x, y) = slist.to_input_response() self.assertEquals(len(x), 2) self.assertEquals(len(y), 2) self.assertTrue(np.array_equal(x[0], slist[0].numpy_array))
def test_from_folder(self): slist = SampleList.from_folder("testdataset/cat1/", [1]) self.assertEquals(len(slist), 2) sample = slist[0] self.assertEquals(sample.tag, [1])
def test_load_data(self): slist = SampleList.from_folder("testdataset/cat1/", [1]) slist.load_data() sample = slist[0] self.assertIsNotNone(sample.numpy_array)