def test_traverse_from_collect_if():
graphA = CoachingGraph()
graphA.init_semi_random_connected(100)
actualA = len(graphA.users)
network = graphTraverser.get_connected_network(graphA.users[0])
sizeA = len(network)
assert sizeA == actualA
def test_get_subgraphs_connected():
graph = CoachingGraph()
graph.init_semi_random_connected(100)
subgraphs = graphTraverser.get_subgraphs(graph.users)
combined = []
for subgraph in subgraphs:
combined.extend(subgraph)
assert set(combined) == set(graph.users)
def test_traverse_from_collect_if(): graphA = CoachingGraph() graphA.init_semi_random_connected(100) actualA = len(graphA.users) network = graphTraverser.get_connected_network(graphA.users[0]) sizeA = len(network) assert sizeA == actualA
def test_get_subgraphs_connected(): graph = CoachingGraph() graph.init_semi_random_connected(100) subgraphs = graphTraverser.get_subgraphs(graph.users) combined = [] for subgraph in subgraphs: combined.extend(subgraph) assert set(combined) == set(graph.users)
def test_infect_limited_infects_correct_number_connected(): graph = CoachingGraph() graph.init_semi_random_connected(200) numToInfect = 70 users = graph.users infecter = Infecter() infecter.infect_limited(users, numToInfect, 0, False) numInfected = infecter.get_num_infected(users) assert numInfected == numToInfect
def test_infect_limited_from(): graph = CoachingGraph() graph.init_semi_random_connected(200) numToInfect = 70 users = graph.users user = random.choice(users) infecter = Infecter() infecter.infect_limited_from(user, numToInfect) numInfected = infecter.get_num_infected(users) assert numInfected == numToInfect
def generate_test_graphs_connected(dirName, numGraphs, maxSize):
logger.info("generating test graphs in directory: {}".format(dirName))
folderPath = refresh_testData_folder(dirName)
for i in range(numGraphs):
graph = CoachingGraph()
fileNumber = i + 1
fileName = '{}_'.format(fileNumber) + TEST_GRAPH_SUFFIX
randomSize = random.randint(40, maxSize)
graph.init_semi_random_connected(randomSize)
filePath = os.path.join(folderPath, fileName)
graph.save_as(filePath)
def generate_test_graphs_connected(dirName, numGraphs, maxSize):
logger.info("generating test graphs in directory: {}".format(dirName))
folderPath = refresh_testData_folder(dirName)
for i in range(numGraphs):
graph = CoachingGraph()
fileNumber = i + 1
fileName = '{}_'.format(fileNumber) + TEST_GRAPH_SUFFIX
randomSize = random.randint(40, maxSize)
graph.init_semi_random_connected(randomSize)
filePath = os.path.join(folderPath, fileName)
graph.save_as(filePath)
def test_init_semi_random_connected():
graph = CoachingGraph()
graph.init_semi_random_connected(200)
numSubgraphs = len(graphTraverser.get_subgraphs(graph.users))
assert numSubgraphs == 1
def test_init_semi_random_connected(): graph = CoachingGraph() graph.init_semi_random_connected(200) numSubgraphs = len(graphTraverser.get_subgraphs(graph.users)) assert numSubgraphs == 1
