formatted using yapf (9fcb6889) · Commits · 钟慕尧 / deepchem

deepchem/splits/tests/test_splitter.py

+2 −3

Original line number	Diff line number	Diff line
		@@ -504,11 +504,11 @@ class TestSplitters(unittest.TestCase):
		# that have no hits.
		assert len(np.where(~w.any(axis=1))[0]) == 0


		def test_indice_split(self):

		solubility_dataset = dc.data.tests.load_solubility_data()
		random_splitter = dc.splits.IndiceSplitter(valid_indices=[7],test_indices=[8])
		random_splitter = dc.splits.IndiceSplitter(
		valid_indices=[7], test_indices=[8])
		train_data, valid_data, test_data = \
		random_splitter.split(
		solubility_dataset)
		@@ -517,7 +517,6 @@ class TestSplitters(unittest.TestCase):
		assert len(test_data) == 1



		if __name__ == "__main__":
		import nose