Wrap single-argument methods
parent
10628a0ddc
commit
a1b8f52da8
|
@ -79,6 +79,8 @@ virtual class DiscretePrior : gtsam::DiscreteConditional {
|
|||
void print(string s = "Discrete Prior\n",
|
||||
const gtsam::KeyFormatter& keyFormatter =
|
||||
gtsam::DefaultKeyFormatter) const;
|
||||
double operator()(size_t value) const;
|
||||
std::vector<double> pmf() const;
|
||||
};
|
||||
|
||||
#include <gtsam/discrete/DiscreteBayesNet.h>
|
||||
|
|
|
@ -13,16 +13,18 @@ Author: Varun Agrawal
|
|||
|
||||
import unittest
|
||||
|
||||
from gtsam import DiscretePrior, DecisionTreeFactor, DiscreteKeys
|
||||
import numpy as np
|
||||
from gtsam import DecisionTreeFactor, DiscreteKeys, DiscretePrior
|
||||
from gtsam.utils.test_case import GtsamTestCase
|
||||
|
||||
X = 0, 2
|
||||
|
||||
|
||||
class TestDiscretePrior(GtsamTestCase):
|
||||
"""Tests for Discrete Priors."""
|
||||
|
||||
def test_constructor(self):
|
||||
"""Test various constructors."""
|
||||
X = 0, 2
|
||||
actual = DiscretePrior(X, "2/3")
|
||||
keys = DiscreteKeys()
|
||||
keys.push_back(X)
|
||||
|
@ -30,10 +32,19 @@ class TestDiscretePrior(GtsamTestCase):
|
|||
expected = DiscretePrior(f)
|
||||
self.gtsamAssertEquals(actual, expected)
|
||||
|
||||
def test_operator(self):
|
||||
prior = DiscretePrior(X, "2/3")
|
||||
self.assertAlmostEqual(prior(0), 0.4)
|
||||
self.assertAlmostEqual(prior(1), 0.6)
|
||||
|
||||
def test_pmf(self):
|
||||
prior = DiscretePrior(X, "2/3")
|
||||
expected = np.array([0.4, 0.6])
|
||||
np.testing.assert_allclose(expected, prior.pmf())
|
||||
|
||||
def test_markdown(self):
|
||||
"""Test the _repr_markdown_ method."""
|
||||
|
||||
X = 0, 2
|
||||
prior = DiscretePrior(X, "2/3")
|
||||
expected = " $P(0)$:\n" \
|
||||
"|0|value|\n" \
|
||||
|
|
Loading…
Reference in New Issue