def test_empty(self):
valid, msg = validate_distributions([[]])
assert isinstance(msg, str)
print("\n",
"While passing empty data, the error messages:",
msg,
end="\n")
assert not valid
def test_ndim_3(self):
data = np.linspace(0, 1, 12).reshape((2, 2, -1))
valid, msg = validate_distributions(data)
print("\n",
"While passing three-dimensional data, the error messages:",
msg,
end="\n")
assert not valid
def test_type_str(self):
valid, array_or_msg = validate_distributions(
self.valid_data.astype(str).tolist())
assert valid
assert isinstance(array_or_msg, np.ndarray)
assert array_or_msg.ndim == 2
assert array_or_msg.shape == self.valid_data.shape
assert np.all(np.less(np.abs(array_or_msg - self.valid_data), 1e-4))
def test_has_nan(self):
data = self.valid_data.copy()
data[:3, -5:] = np.nan
valid, msg = validate_distributions(data)
print("\n",
"While it has NaN values, the error messages:",
msg,
end="\n")
assert not valid
def test_not_sum_to_one(self):
data = self.valid_data.copy()
data[:3, -5:] = 1e3
valid, msg = validate_distributions(data)
print(
"\n",
"While the sums of some rows are not equal to 1, the error messages:",
msg,
end="\n")
assert not valid
def test_type_none(self):
valid, msg = validate_distributions(None)
assert isinstance(msg, str)
print("\n", "While passing `None`, the error messages:", msg, end="\n")
assert not valid
def test_with_int(self):
with_int = self.valid_data.tolist()
with_int[0][1] += with_int[0][0]
with_int[0][0] = 0
valid, _ = validate_distributions(with_int)
assert valid
def test_list(self):
valid, _ = validate_distributions(self.valid_data.tolist())
assert valid
def test_sliced(self):
valid, _ = validate_distributions(self.valid_data[:-1])
assert valid
def test_np_float64(self):
valid, _ = validate_distributions(self.valid_data.astype(np.float64))
assert valid
def test_valid(self):
valid, array_or_msg = validate_distributions(self.valid_data.tolist())
assert valid
assert isinstance(array_or_msg, np.ndarray)
assert array_or_msg.ndim == 2
assert np.all(np.less(np.sum(array_or_msg, axis=1) - 1.0, 1e-4))
