Python CoachingGraph.init_semi_random Examples

Example #1

def test_no_user_has_self_as_neighbor():
    graph = CoachingGraph()
    graph.init_semi_random(100)
    for user in graph.users:
        if user in user.neighbors:
            assert False
    assert True

Example #2

File: test_coachingGraph.py Project: lyonsno/infectLimited

def test_no_user_has_self_as_neighbor():
	graph = CoachingGraph()
	graph.init_semi_random(100)
	for user in graph.users:
		if user in user.neighbors:
			assert False
	assert True

Example #3

def test_save_and_load():
    graph = CoachingGraph()
    graph.init_semi_random(100)
    graph.save_as('save_and_load')
    graphB = CoachingGraph()
    graphB.load_from('save_and_load')

    assert len(graph.users) == len(graphB.users)

Example #4

File: test_coachingGraph.py Project: lyonsno/infectLimited

def test_save_and_load():
	graph = CoachingGraph()
	graph.init_semi_random(100)
	graph.save_as('save_and_load')
	graphB = CoachingGraph()
	graphB.load_from('save_and_load')

	assert len(graph.users) == len(graphB.users)

Example #5

def test_get_num_infected_large():
	expectedInfected = 100
	graph = CoachingGraph()
	graph.init_semi_random(expectedInfected)
	infecter = Infecter()

	infecter.infect_all(graph.users)
	actualInfected = infecter.get_num_infected(graph.users)

	assert actualInfected == expectedInfected

Example #6

def test_infect_limited_doesnt_change_num_users():
	originalUsers = 100
	graph = CoachingGraph()
	graph.init_semi_random(originalUsers)
	infecter = Infecter()

	infecter.infect_limited(graph.users, 50, 10)
	remaingingUsers = len(graph.users)

	assert remaingingUsers == originalUsers

Example #7

def test_infect_limited_infects_correct_number():
	graph = CoachingGraph()
	graph.init_semi_random(200)
	numToInfect = 70
	users = graph.users
	infecter = Infecter()

	infecter.infect_limited(users, numToInfect, 0, False)
	numInfected = infecter.get_num_infected(users)

	assert numInfected == numToInfect

Example #8

File: utils.py Project: lyonsno/infectLimited

def generate_test_graphs(dirName, numGraphs, maxSize):
	logger.info("generating test graphs in directory: {}".format(dirName))
	folderPath = refresh_testData_folder(dirName)
	graphs = []
	for i in range(numGraphs):
		graph = CoachingGraph()

		fileNumber = i + 1
		fileName = '{}_'.format(fileNumber) + TEST_GRAPH_SUFFIX

		randomSize = random.randint(100, maxSize)
		randomInternalConnections = random.randint(2, 12)
		randomCoacheesFactor = random.randint(2, 5)
		graph.init_semi_random(200, randomInternalConnections, randomCoacheesFactor)
		graphs.append(graph)
		filePath = os.path.join(folderPath, fileName)
		graph.save_as(filePath)
	return graphs

Example #9

def generate_test_graphs(dirName, numGraphs, maxSize):
    logger.info("generating test graphs in directory: {}".format(dirName))
    folderPath = refresh_testData_folder(dirName)
    graphs = []
    for i in range(numGraphs):
        graph = CoachingGraph()

        fileNumber = i + 1
        fileName = '{}_'.format(fileNumber) + TEST_GRAPH_SUFFIX

        randomSize = random.randint(100, maxSize)
        randomInternalConnections = random.randint(2, 12)
        randomCoacheesFactor = random.randint(2, 5)
        graph.init_semi_random(200, randomInternalConnections,
                               randomCoacheesFactor)
        graphs.append(graph)
        filePath = os.path.join(folderPath, fileName)
        graph.save_as(filePath)
    return graphs

Example #10

def resource_semi_random_graph_users():
	graph = CoachingGraph()
	graph.init_semi_random(200)

	return graph.users

Example #11

def test_semi_random_graph():
    graph = CoachingGraph()
    graph.init_semi_random(20)

    assert len(graph.users) == 20

Example #12

File: test_coachingGraph.py Project: lyonsno/infectLimited

def test_semi_random_graph():
	graph = CoachingGraph()
	graph.init_semi_random(20)

	assert len(graph.users) == 20