def test_getClassTrajWeight(self):
trajs = {}
trajs["a"] = {}
trajs["b"] = {}
trajs["c"] = {}
trajs["a"]["species"] = "hey"
trajs["b"]["species"] = "ho"
trajs["c"]["species"] = "hey"
trajs["a"]["length"] = 100
trajs["b"]["length"] = 200
trajs["c"]["length"] = 400
self.assertEqual(Ensemble.getClassTrajWeight("a", trajs), 1. / 5)
def test_getClassTrajWeight(self):
trajs = {}
trajs["a"] = {}
trajs["b"] = {}
trajs["c"] = {}
trajs["a"]["species"] = "hey"
trajs["b"]["species"] = "ho"
trajs["c"]["species"] = "hey"
trajs["a"]["length"] = 100
trajs["b"]["length"] = 200
trajs["c"]["length"] = 400
self.assertEqual(Ensemble.getClassTrajWeight("a", trajs), 1. / 5)
def test_getClassTrajSamples(self):
trajs = {}
trajs["a"] = {}
trajs["b"] = {}
trajs["c"] = {}
trajs["a"]["species"] = "hey"
trajs["b"]["species"] = "ho"
trajs["c"]["species"] = "hey"
trajs["a"]["length"] = 100
trajs["b"]["length"] = 200
trajs["c"]["length"] = 400
self.assertEqual(Ensemble.getClassTrajSamples("hey", trajs), 500)
def test_getClassTrajSamples(self):
trajs = {}
trajs["a"] = {}
trajs["b"] = {}
trajs["c"] = {}
trajs["a"]["species"] = "hey"
trajs["b"]["species"] = "ho"
trajs["c"]["species"] = "hey"
trajs["a"]["length"] = 100
trajs["b"]["length"] = 200
trajs["c"]["length"] = 400
self.assertEqual(Ensemble.getClassTrajSamples("hey", trajs), 500)
