def generate(num_rows):
ii32 = np.iinfo(np.int32)
values = np.random.randint(ii32.min, ii32.max, size=2)
additional_values = np.full(num_rows - 2, values[1]).astype(np.float)
array = np.concatenate([values, add_nans(additional_values)])
np.random.shuffle(array)
return array
def test_add_nulls_int():
array = np.arange(100)
with_nans = utils.add_nans(array)
assert len(with_nans) == 100
assert 1 <= np.isnan(with_nans).sum() < 100
nans = np.isnan(with_nans)
np.testing.assert_array_equal(array[~nans], with_nans[~nans])
def generate(num_rows):
return add_nans(
ConstantIntegerGenerator.generate(num_rows).astype(np.float))
def generate(num_rows):
return add_nans(
RandomIntegerGenerator.generate(num_rows).astype(np.float))
def generate(num_rows):
return add_nans(BigNormalGenerator.generate(num_rows))
def generate(num_rows):
dates = RandomGapDatetimeGenerator.generate(num_rows)
return add_nans(dates.astype('O'))
def generate(num_rows):
return add_nans(RandomStringGenerator.generate(num_rows).astype('O'))
def generate(num_rows):
return add_nans(UniqueStringGenerator.generate(num_rows).astype('O'))
def generate(num_rows):
return add_nans(UniqueIntegerGenerator.generate(num_rows))