def test_cos_dataset():
 skip_if_no_data()
 dataset = CosDataset()
 sample_batch = dataset.get_batch_design(batch_size=10000)
 assert (sample_batch.shape == (10000, 2))
 assert (sample_batch[:, 0].min() >= dataset.min_x)
 assert (sample_batch[:, 0].max() <= dataset.max_x)

def test_cos_dataset ( ) : skip_if_no_data ( ) dataset = CosDataset ( ) sample_batch = dataset . get_batch_design ( batch_size = 10000 ) assert ( sample_batch . shape == ( 10000 , 2 ) ) assert ( sample_batch [ : , 0 ] . min ( ) >= dataset . min_x ) assert ( sample_batch [ : , 0 ] . max ( ) <= dataset . max_x )
tests if the dataset generator yields the desired value .