Wrap DiscretePrior
							parent
							
								
									268a49ec1c
								
							
						
					
					
						commit
						4727783304
					
				|  | @ -71,6 +71,16 @@ virtual class DiscreteConditional : gtsam::DecisionTreeFactor { | |||
|                  gtsam::DefaultKeyFormatter) const; | ||||
| }; | ||||
| 
 | ||||
| #include <gtsam/discrete/DiscretePrior.h> | ||||
| virtual class DiscretePrior : gtsam::DiscreteConditional { | ||||
|   DiscretePrior(); | ||||
|   DiscretePrior(const gtsam::DecisionTreeFactor& f); | ||||
|   DiscretePrior(const gtsam::DiscreteKey& key, string spec); | ||||
|   void print(string s = "Discrete Prior\n", | ||||
|              const gtsam::KeyFormatter& keyFormatter = | ||||
|                  gtsam::DefaultKeyFormatter) const; | ||||
| }; | ||||
| 
 | ||||
| #include <gtsam/discrete/DiscreteBayesNet.h> | ||||
| class DiscreteBayesNet {  | ||||
|   DiscreteBayesNet(); | ||||
|  |  | |||
|  | @ -0,0 +1,49 @@ | |||
| """ | ||||
| GTSAM Copyright 2010-2021, Georgia Tech Research Corporation, | ||||
| Atlanta, Georgia 30332-0415 | ||||
| All Rights Reserved | ||||
| 
 | ||||
| See LICENSE for the license information | ||||
| 
 | ||||
| Unit tests for Discrete Priors. | ||||
| Author: Varun Agrawal | ||||
| """ | ||||
| 
 | ||||
| # pylint: disable=no-name-in-module, invalid-name | ||||
| 
 | ||||
| import unittest | ||||
| 
 | ||||
| from gtsam import DiscretePrior, DecisionTreeFactor, DiscreteKeys | ||||
| from gtsam.utils.test_case import GtsamTestCase | ||||
| 
 | ||||
| 
 | ||||
| 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) | ||||
|         f = DecisionTreeFactor(keys, "0.4 0.6") | ||||
|         expected = DiscretePrior(f) | ||||
|         self.gtsamAssertEquals(actual, expected) | ||||
| 
 | ||||
|     def test_markdown(self): | ||||
|         """Test the _repr_markdown_ method.""" | ||||
| 
 | ||||
|         X = 0, 2 | ||||
|         prior = DiscretePrior(X, "2/3") | ||||
|         expected = " $P(0)$:\n" \ | ||||
|             "|0|value|\n" \ | ||||
|             "|:-:|:-:|\n" \ | ||||
|             "|0|0.4|\n" \ | ||||
|             "|1|0.6|\n" \ | ||||
| 
 | ||||
|         actual = prior._repr_markdown_() | ||||
|         self.assertEqual(actual, expected) | ||||
| 
 | ||||
| 
 | ||||
| if __name__ == "__main__": | ||||
|     unittest.main() | ||||
		Loading…
	
		Reference in New Issue