class UnitTests(unittest.TestCase): def setUp(self): from netica import Netica self.netica = Netica() self.netfilename = 'curvnet.dne' self.env = self.netica.newenv() self.netica.initenv(self.env) self.net = self.netica.opennet(self.env, self.netfilename) self.nl_p = self.netica.getnetnodes(self.net) self.netica.compilenet(self.net) def test_enternodelikelyhood(self): self.netica.retractnetfindings(self.net) node_p = self.netica.getnodenamed('radius', self.net) self.netica.enternodelikelyhood(node_p, [1, 0, 0, 0, 0, 0]) # self.netica.savenet(env, net, 'test.dne') def test_equationtotable(self): node_p = self.netica.getnodenamed('radius', self.net) self.netica.equationtotable(node_p, 10, True, True) def test_revisecptsbycasefile(self): casefilename = 'wavedir.cas' file_p = self.netica._newstream(self.env, casefilename) self.netica.revisecptsbycasefile(file_p, self.nl_p, 0, 1) def test_getnodeequation(self): node_p = self.netica.getnodenamed('phi', self.net) eqn = self.netica.getnodeequation(node_p) print eqn # the radius node has no equation, thus should result in an empty string node_p = self.netica.getnodenamed('radius', self.net) eqn = self.netica.getnodeequation(node_p) self.assertEqual(eqn, '') def test_setnodeequation(self): node_p = self.netica.getnodenamed('phi', self.net) eqn_in = 'phi (theta, wavedir) = theta - wavedir' self.netica.setnodeequation(node_p, eqn_in) eqn_out = self.netica.getnodeequation(node_p) self.assertEqual(eqn_in, eqn_out) def test_newnode(self): nodename = 'test' node_p = self.netica.newnode(nodename, 3, self.net) self.assertEqual(self.netica.getnodename(node_p), nodename) def test_deletenode(self): nodename = 'test' node_p = self.netica.newnode(nodename, 3, self.net) self.netica.deletenode(node_p) def test_addlink(self): parent = self.netica.getnodenamed('wavedir', self.net) child = self.netica.getnodenamed('erosionfactor', self.net) link_index = self.netica.addlink(parent, child) print 'index', link_index def test_deletelink(self): child = self.netica.getnodenamed('erosionfactor', self.net) link_index = 1 self.netica.deletelink(link_index, child) def test_getnodelevels_continuous(self): node_p = self.netica.getnodenamed('phi', self.net) print self.netica.getnodelevels(node_p) def test_getnodelevels_discrete(self): node_p = self.netica.newnode('Time_horizon', 3, self.net) #self.netica.setnodestatenames(node_p, "one, five, ten") self.netica.getnodelevels(node_p) def test_getnodestatename(self): node_p = self.netica.newnode('Time_horizon', 3, self.net) self.netica.setnodestatenames(node_p, "one, five, ten") print 'STATE: "' + self.netica.getnodestatename(node_p, 0) + '"' def test_enterfinding(self): node_p = self.netica.getnodenamed('phi', self.net) self.netica.enterfinding(node_p, 0) def test_getnodeexpectedvalue(self): node_p = self.netica.getnodenamed('phi', self.net) expval = self.netica.getnodeexpectedvalue(node_p) print 'expval = ', expval
class UnitTests(unittest.TestCase): def setUp(self): from netica import Netica self.netica = Netica() self.netfilename = 'curvnet.dne' self.env = self.netica.newenv() self.netica.initenv(self.env) self.net = self.netica.opennet(self.env, self.netfilename) self.nl_p = self.netica.getnetnodes(self.net) self.netica.compilenet(self.net) def test_enternodelikelyhood(self): self.netica.retractnetfindings(self.net) node_p = self.netica.getnodenamed('radius', self.net) self.netica.enternodelikelyhood(node_p, [1,0,0,0,0,0]) # self.netica.savenet(env, net, 'test.dne') def test_equationtotable(self): node_p = self.netica.getnodenamed('radius', self.net) self.netica.equationtotable(node_p, 10, True, True) def test_revisecptsbycasefile(self): casefilename = 'wavedir.cas' file_p = self.netica._newstream(self.env, casefilename) self.netica.revisecptsbycasefile(file_p, self.nl_p, 0, 1) def test_getnodeequation(self): node_p = self.netica.getnodenamed('phi', self.net) eqn = self.netica.getnodeequation(node_p) print eqn # the radius node has no equation, thus should result in an empty string node_p = self.netica.getnodenamed('radius', self.net) eqn = self.netica.getnodeequation(node_p) self.assertEqual(eqn, '') def test_setnodeequation(self): node_p = self.netica.getnodenamed('phi', self.net) eqn_in = 'phi (theta, wavedir) = theta - wavedir' self.netica.setnodeequation(node_p, eqn_in) eqn_out = self.netica.getnodeequation(node_p) self.assertEqual(eqn_in, eqn_out) def test_newnode(self): nodename = 'test' node_p = self.netica.newnode(nodename, 3, self.net) self.assertEqual(self.netica.getnodename(node_p), nodename) def test_deletenode(self): nodename = 'test' node_p = self.netica.newnode(nodename, 3, self.net) self.netica.deletenode(node_p) def test_addlink(self): parent = self.netica.getnodenamed('wavedir', self.net) child = self.netica.getnodenamed('erosionfactor', self.net) link_index = self.netica.addlink(parent, child) print 'index',link_index def test_deletelink(self): child = self.netica.getnodenamed('erosionfactor', self.net) link_index = 1 self.netica.deletelink(link_index, child) def test_getnodelevels_continuous(self): node_p = self.netica.getnodenamed('phi', self.net) print self.netica.getnodelevels(node_p) def test_getnodelevels_discrete(self): node_p = self.netica.newnode('Time_horizon', 3, self.net) #self.netica.setnodestatenames(node_p, "one, five, ten") self.netica.getnodelevels(node_p) def test_getnodestatename(self): node_p = self.netica.newnode('Time_horizon', 3, self.net) self.netica.setnodestatenames(node_p, "one, five, ten") print 'STATE: "' + self.netica.getnodestatename(node_p, 0) + '"' def test_enterfinding(self): node_p = self.netica.getnodenamed('phi', self.net) self.netica.enterfinding(node_p, 0) def test_getnodeexpectedvalue(self): node_p = self.netica.getnodenamed('phi', self.net) expval = self.netica.getnodeexpectedvalue(node_p) print 'expval = ',expval