modify shapes expected by unit test for FASTA loader (9ae8afcd) · Commits · 钟慕尧 / deepchem

deepchem/data/tests/test_fasta_loader.py

+5 −0

Original line number	Diff line number	Diff line
		@@ -17,6 +17,8 @@ class TestFASTALoader(unittest.TestCase):
		self.current_dir = os.path.dirname(os.path.abspath(__file__))

		def test_fasta_load(self):
		legacy = False

		input_file = os.path.join(self.current_dir,
		"../../data/tests/example.fasta")
		loader = dc.data.FASTALoader()
		@@ -25,4 +27,7 @@ class TestFASTALoader(unittest.TestCase):
		# example.fasta contains 3 sequences each of length 58.
		# The one-hot encoding turns base-pairs into vectors of length 5 (ATCGN).
		# There is one "image channel".
		if (legacy == True):
		assert sequences.X.shape == (3, 5, 58, 1)
		else:
		assert sequences.X.shape == (3, 58, 6)