finished tests -- passes

        required_hit_dict = self.__generate_required_hits(y_df, frac_train)

        index_dict = self.__generate_required_index(y_df, required_hit_dict)

        print(index_dict)

        X_train, X_test, y_train, y_test, w_train, w_test, id_train, id_test = [], [], [], [], [], [], [], []

        # cycle through rows in y, copy over rows as appropriate

"""

Tests for splitter objects.

"""

from __future__ import print_function

from __future__ import division

from __future__ import print_function

from __future__ import unicode_literals

__author__ = "Bharath Ramsundar"

__copyright__ = "Copyright 2016, Stanford University"

__license__ = "GPL"

import os

import unittest

from deepchem.splits import RandomSplitter

from deepchem.splits import ScaffoldSplitter

from deepchem.splits import StratifiedSplitter

from deepchem.datasets.tests import TestDatasetAPI

from deepchem.datasets import Dataset

import pandas as pd

        assert len(test_data) == 1

    def test_stratified_multitask_split(self):

        print("In stratified tester")

        """

        Test multitask StratifiedSplitter class

        """

        sparse_dataset = self.load_sparse_multitask_dataset()

        stratified_splitter = StratifiedSplitter()

        train_data, valid_data, test_data = \