Move tests > 50 seconds to jenkins (23d92453) · Commits · 钟慕尧 / deepchem

.travis.yml

+1 −1

Original line number	Diff line number	Diff line
		@@ -20,7 +20,7 @@ install:
		- pip install coveralls
		- python setup.py install
		script:
		- nosetests --with-timer --with-coverage --cover-package=deepchem -v deepchem --nologcapture
		- nosetests -a '!slow' --with-timer --with-coverage --cover-package=deepchem -v deepchem --nologcapture
		- find ./deepchem \| grep .py$ \|xargs python -m doctest -v
		- bash devtools/travis-ci/test_format_code.sh
		after_success:

+3 −4

Original line number	Diff line number	Diff line
		"""
		Tests for Docking
		"""
		from __future__ import print_function
		from __future__ import division
		from __future__ import print_function
		from __future__ import unicode_literals

		__author__ = "Bharath Ramsundar"
		@@ -10,10 +10,8 @@ __copyright__ = "Copyright 2016, Stanford University"
		__license__ = "MIT"

		import unittest
		import tempfile
		import os
		import shutil
		import numpy as np
		from nose.tools import attr
		import sys
		import deepchem as dc

		@@ -117,6 +115,7 @@ class TestDocking(unittest.TestCase):
		assert os.path.exists(protein_docked)
		assert os.path.exists(ligand_docked)

		@attr('slow')
		def test_pocket_vina_grid_dnn_docker_dock(self):
		"""Test that VinaGridDNNDocker can dock."""
		if sys.version_info >= (3, 0):

+2 −4

Original line number	Diff line number	Diff line
		"""
		Tests for Pose Generation
		"""
		from __future__ import print_function
		from __future__ import division
		from __future__ import print_function
		from __future__ import unicode_literals

		__author__ = "Bharath Ramsundar"
		@@ -12,9 +12,6 @@ __license__ = "MIT"
		import os
		import sys
		import unittest
		import tempfile
		import shutil
		import numpy as np
		import deepchem as dc


		@@ -50,6 +47,7 @@ class TestPoseGeneration(unittest.TestCase):
		assert os.path.exists(protein_pose_file)
		assert os.path.exists(ligand_pose_file)

		@attr('slow')
		def test_pocket_vina_poses(self):
		"""Test that VinaPoseGenerator creates pose files."""
		if sys.version_info >= (3, 0):

+2 −1

Original line number	Diff line number	Diff line
		from unittest import TestCase

		from nose.tools import assert_equals
		from nose.tools import assert_equals, attr
		from rdkit import Chem

		import deepchem as dc
		@@ -11,6 +11,7 @@ from deepchem.models.autoencoder_models.autoencoder import TensorflowMoleculeEnc

		class TestTensorflowEncoders(TestCase):

		@attr('slow')
		def test_fit(self):
		tf_enc = TensorflowMoleculeEncoder.zinc_encoder()

+9 −5

Original line number	Diff line number	Diff line
		"""
		Tests for molnet function
		"""
		from __future__ import print_function
		from __future__ import division
		from __future__ import print_function
		from __future__ import unicode_literals

		import os
		import csv
		import tempfile
		import unittest

		import numpy as np
		import pandas as pd
		import os
		from nose.tools import attr

		import deepchem as dc
		import tempfile
		import csv


		class TestMolnet(unittest.TestCase):
		@@ -23,6 +25,7 @@ class TestMolnet(unittest.TestCase):
		super(TestMolnet, self).setUp()
		self.current_dir = os.path.dirname(os.path.abspath(__file__))

		@attr('slow')
		def test_delaney_graphconvreg(self):
		"""Tests molnet benchmarking on delaney with graphconvreg."""
		datasets = ['delaney']
		@@ -45,6 +48,7 @@ class TestMolnet(unittest.TestCase):
		assert float(lastrow[-3]) > 0.75
		os.remove(os.path.join(out_path, 'results.csv'))

		@attr('slow')
		def test_qm7_multitask(self):
		"""Tests molnet benchmarking on qm7 with multitask network."""
		datasets = ['qm7']