示例#1
0
文件: f2.py 项目: anvesh2502/Leetcode
# Test 1 : Check for FCM creation
fcm_graph=FCM()

print '### FCM creation successful'

# Test 2 : Test for adding concepts
fcm_graph.add_concept("concept1")
fcm_graph.add_concept("concept2")

print '### FCM adding valid concepts successful'


# Test 3 : Adding valid edges

fcm_graph.add_edge('concept1','concept2',0.3)

print '### FCM adding valid edges successful'

fcm_graph.remove_edge('concept1','concept2')
print '### FCM removing valid edges successful'


# Test 3 : Adding edges which did not exist

try :
  fcm_graph.add_edge('new concept1','new concept2',0.3)
except ConceptExistError :
  print '### FCM removing adding invalid edges successful '

test_fcm.add_concept("Heat element")
test_fcm.add_concept("Therm_tank1")
test_fcm.set_value("Tank1", .2)
test_fcm.add_concept("Therm_tank2")

test_fcm.set_value("Tank2", .01)

test_fcm.set_value("Valve1", .55)
test_fcm.set_value("Valve2", .58)
test_fcm.set_value("Valve3", .0)

test_fcm.set_value("Heat element", .2)
test_fcm.set_value("Therm_tank1", .1)
test_fcm.set_value("Therm_tank2", .05)

test_fcm.add_edge("Tank1", "Valve1", .21)
test_fcm.add_edge("Tank1", "Valve2", .38)
test_fcm.add_edge("Tank2", "Valve2", .7)
test_fcm.add_edge("Tank2", "Valve3", .6)
test_fcm.add_edge("Valve1", "Tank1", .76)
test_fcm.add_edge("Valve2", "Tank1", -.6)
test_fcm.add_edge("Valve2", "Tank2", .8)
test_fcm.add_edge("Valve2", "Therm_tank2", .09)
test_fcm.add_edge("Valve3", "Tank2", -.42)
test_fcm.add_edge("Heat element", "Therm_tank1", .6)
test_fcm.add_edge("Therm_tank1", "Heat element", .53)
test_fcm.add_edge("Therm_tank1", "Valve1", .4)
test_fcm.add_edge("Therm_tank2", "Valve2", .3)

pso = PSO(test_fcm, {'Tank1': (0.6, 0.9), 'Tank2': (0.3, 0.9)})
pso.run_convergence()