Commit 32fef346 authored by peastman's avatar peastman
Browse files

Removed test that didn't really load the dataset it claimed to

parent 31dd8d58
Loading
Loading
Loading
Loading
+16 −17
Original line number Diff line number Diff line
@@ -6,20 +6,19 @@ import os
import numpy as np
from deepchem.molnet import load_qm9


def test_qm9_loader():
  current_dir = os.path.dirname(os.path.abspath(__file__))
  tasks, datasets, transformers = load_qm9(
      reload=False,
      data_dir=current_dir,
      featurizer='ECFP',
      splitter_kwargs={
          'seed': 42,
          'frac_train': 0.6,
          'frac_valid': 0.2,
          'frac_test': 0.2
      })

  assert len(tasks) == 12
  assert tasks[0] == 'mu'
  assert datasets[0].X.shape == (8, 1024)
# def test_qm9_loader():
#   current_dir = os.path.dirname(os.path.abspath(__file__))
#   tasks, datasets, transformers = load_qm9(
#       reload=False,
#       data_dir=current_dir,
#       featurizer='ECFP',
#       splitter_kwargs={
#           'seed': 42,
#           'frac_train': 0.6,
#           'frac_valid': 0.2,
#           'frac_test': 0.2
#       })
#
#   assert len(tasks) == 12
#   assert tasks[0] == 'mu'
#   assert datasets[0].X.shape == (8, 1024)