def test_can_pass_scalar_to_vertex() -> None: gaussian = Gaussian(0., 1.) sample = gaussian.sample() assert type(sample) == numpy_types assert sample.shape == () assert sample.dtype == float
def test_vertex_sample_is_a_numpy_array() -> None: mu = np.array([[1., 2.], [3., 4.]]) sigma = np.array([[.1, .2], [.3, .4]]) vertex = Gaussian(mu, sigma) value = vertex.sample() assert type(value) == np.ndarray assert value.dtype == np.float64 assert value.shape == (2, 2)
def test_can_pass_pandas_series_to_vertex() -> None: gaussian = Gaussian(pd.Series(data=[0.1, 0.4]), pd.Series(data=[0.1, 0.4])) sample = gaussian.sample() assert sample.shape == (2, )
def test_can_pass_pandas_dataframe_to_vertex() -> None: gaussian = Gaussian(pd.DataFrame(data=[0.1, 0.4]), pd.DataFrame(data=[0.1, 0.4])) sample = gaussian.sample() assert sample.shape == (2, 1)
def test_can_pass_ndarray_to_vertex() -> None: gaussian = Gaussian(np.array([0.1, 0.4]), np.array([0.4, 0.5])) sample = gaussian.sample() assert sample.shape == (2, )