Commit 62714121 authored by VIGNESHinZONE's avatar VIGNESHinZONE
Browse files

fixing slow test

parent c6f70e74
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -103,6 +103,7 @@ class TestConvMolFeaturizer(unittest.TestCase):
class TestAtomicConvFeaturizer(unittest.TestCase):

  @pytest.mark.slow
  @pytest.mark.tensorflow
  def test_feature_generation(self):
    """Test if featurization works using AtomicConvFeaturizer."""
    dir_path = os.path.dirname(os.path.realpath(__file__))
+2 −0
Original line number Diff line number Diff line
@@ -28,6 +28,7 @@ class TestMolnet(unittest.TestCase):
    self.current_dir = os.path.dirname(os.path.abspath(__file__))

  @pytest.mark.slow
  @pytest.mark.tensorflow
  def test_delaney_graphconvreg(self):
    """Tests molnet benchmarking on delaney with graphconvreg."""
    datasets = ['delaney']
@@ -46,6 +47,7 @@ class TestMolnet(unittest.TestCase):
    os.remove(os.path.join(out_path, 'results.csv'))

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