Python load_tests_factory示例

示例#1

文件： test.py 项目： M4573R/discopt

#! /usr/bin/env python3


import sys
sys.path.insert(0, '..')
import test_answers as ta
import solver


class TestTSP(ta.TestAnswers):

	PREFIX = 'tsp'
	PROBLEMS = ((50, 3), (70, 7), (100, 5), (250, 9), (500, 1), (1000, 5))
	EXPECTED_ANSWERS = (428, 20750, 29440, 36934, 316527, 67217770) # The last is wrong

	@staticmethod
	def solve(data):
		return solver.solveIt(data, time=True, assertions=True)


load_tests = ta.load_tests_factory(TestTSP)


if __name__ == '__main__':
	ta.main()

示例#2

文件： test.py 项目： M4573R/discopt

#! /usr/bin/env python3


import sys
sys.path.insert(0, '..')
import test_answers as ta
import solver


class TestColoring(ta.TestAnswers):

	PREFIX = 'gc'
	PROBLEMS = ((50, 3), (70, 7), (100, 5), (250, 9), (500, 1), (1000, 5))
	EXPECTED_ANSWERS = (6, 17, 15, 73, 12, 85) # The latter one is wrong

	@staticmethod
	def solve(data):
		return solver.solveIt(data, time=True, assertions=True)


load_tests = ta.load_tests_factory(TestColoring)


if __name__ == '__main__':
	ta.main()