diff --git a/gtsam/discrete/DiscretePrior.h b/gtsam/discrete/DiscretePrior.h index 1a7c6ae6c..d11d9be06 100644 --- a/gtsam/discrete/DiscretePrior.h +++ b/gtsam/discrete/DiscretePrior.h @@ -98,7 +98,7 @@ class GTSAM_EXPORT DiscretePrior : public DiscreteConditional { * sample * @return sample from conditional */ - size_t sample() const { return Base::sample({}); } + size_t sample() const { return Base::sample(DiscreteValues()); } /// @} }; diff --git a/python/gtsam/tests/test_DiscretePrior.py b/python/gtsam/tests/test_DiscretePrior.py index 4f017d66a..5bf6a8d19 100644 --- a/python/gtsam/tests/test_DiscretePrior.py +++ b/python/gtsam/tests/test_DiscretePrior.py @@ -6,7 +6,7 @@ All Rights Reserved See LICENSE for the license information Unit tests for Discrete Priors. -Author: Varun Agrawal +Author: Frank Dellaert """ # pylint: disable=no-name-in-module, invalid-name @@ -42,6 +42,11 @@ class TestDiscretePrior(GtsamTestCase): expected = np.array([0.4, 0.6]) np.testing.assert_allclose(expected, prior.pmf()) + def test_sample(self): + prior = DiscretePrior(X, "2/3") + actual = prior.sample() + self.assertIsInstance(actual, int) + def test_markdown(self): """Test the _repr_markdown_ method."""