Commit cae1a4ea authored by Aneesh Pappu's avatar Aneesh Pappu
Browse files

updating sparsity test

parent b5da23dc
Loading
Loading
Loading
Loading
+5 −1
Original line number Diff line number Diff line
@@ -86,9 +86,13 @@ class TestSplitters(TestDatasetAPI):
        Test multitask StratifiedSplitter class
        """
        #ensure sparse dataset is actually sparse

        sparse_dataset = self.load_sparse_multitask_dataset()
        sparse_np_list = sparse_dataset.to_numpy()
        sparse_np = sparse_np_list[1]
        frac_train = 0.8
        cutoff = frac_train * len(sparse_np)
        sparse_np = sparse_np[:cutoff]
        sparse_df = pd.DataFrame(data = sparse_np)
        total_rows = len(sparse_df.index)
        sparse_flag = False
@@ -101,7 +105,7 @@ class TestSplitters(TestDatasetAPI):
                assert NaN_count == total_rows
                break
        if not sparse_flag:
            print("datset isn't sparse")
            print("dataset isn't sparse")
            assert sparse_flag is True
        else:
            print("dataset is sparse")