:bug: fix flake8 error about data loader tests (b2851cad) · Commits · 钟慕尧 / deepchem

deepchem/data/tests/test_csv_loader.py

+0 −2

Original line number	Diff line number	Diff line
		import os
		from io import StringIO
		import tempfile
		import shutil
		import deepchem as dc

deepchem/data/tests/test_data_loader.py

+0 −22

Original line number	Diff line number	Diff line
		@@ -3,7 +3,6 @@ Tests for FeaturizedSamples class
		"""

		import os
		import unittest
		import tempfile
		import shutil
		import deepchem as dc
		@@ -22,13 +21,7 @@ def test_unlabelled():
		def test_scaffold_test_train_valid_test_split():
		"""Test of singletask RF ECFP regression API."""
		current_dir = os.path.dirname(os.path.abspath(__file__))
		splittype = "scaffold"
		input_transforms = []
		output_transforms = ["normalize"]
		model_params = {}
		tasks = ["log-solubility"]
		task_type = "regression"
		task_types = {task: task_type for task in tasks}
		input_file = os.path.join(current_dir, "../../models/tests/example.csv")
		featurizer = dc.feat.CircularFingerprint(size=1024)

		@@ -50,13 +43,7 @@ def test_scaffold_test_train_valid_test_split():
		def test_scaffold_test_train_test_split():
		"""Test of singletask RF ECFP regression API."""
		current_dir = os.path.dirname(os.path.abspath(__file__))
		splittype = "scaffold"
		input_transforms = []
		output_transforms = ["normalize"]
		model_params = {}
		tasks = ["log-solubility"]
		task_type = "regression"
		task_types = {task: task_type for task in tasks}
		input_file = os.path.join(current_dir, "../../models/tests/example.csv")
		featurizer = dc.feat.CircularFingerprint(size=1024)

		@@ -76,12 +63,7 @@ def test_scaffold_test_train_test_split():
		def test_random_test_train_valid_test_split():
		"""Test of singletask RF ECFP regression API."""
		current_dir = os.path.dirname(os.path.abspath(__file__))
		input_transforms = []
		output_transforms = ["normalize"]
		model_params = {}
		tasks = ["log-solubility"]
		task_type = "regression"
		task_types = {task: task_type for task in tasks}
		input_file = os.path.join(current_dir, "../../models/tests/example.csv")
		featurizer = dc.feat.CircularFingerprint(size=1024)

		@@ -103,11 +85,7 @@ def test_random_test_train_valid_test_split():
		def test_random_test_train_test_split():
		"""Test of singletask RF ECFP regression API."""
		current_dir = os.path.dirname(os.path.abspath(__file__))
		#splittype = "random"
		model_params = {}
		tasks = ["log-solubility"]
		task_type = "regression"
		task_types = {task: task_type for task in tasks}
		input_file = os.path.join(current_dir, "../../models/tests/example.csv")
		featurizer = dc.feat.CircularFingerprint(size=1024)
		loader = dc.data.CSVLoader(

deepchem/data/tests/test_fasta_loader.py

+0 −3

Original line number	Diff line number	Diff line
		"""
		Tests that FASTA files can be loaded.
		"""
		__author__ = "Bharath Ramsundar"
		__license__ = "MIT"

		import os
		import unittest

deepchem/data/tests/test_image_loader.py

+0 −1

Original line number	Diff line number	Diff line
		@@ -30,7 +30,6 @@ class TestImageLoader(unittest.TestCase):
		Image.fromarray(self.face).save(self.face_copy_path)

		# Create zip of image file
		#self.zip_path = "/home/rbharath/misc/cells.zip"
		self.zip_path = os.path.join(self.data_dir, "face.zip")
		zipf = zipfile.ZipFile(self.zip_path, "w", zipfile.ZIP_DEFLATED)
		zipf.write(self.face_path)

Admin message