def testExample2(self): self.assertEqual(self.testSubject.run(['1110111011', '10011']), [0]) class ProblemSolver(): def run(self, input): if input[0].find(input[1]) != -1: return [1] else: return [0] # MAIN def runTests(): suite = unittest.TestLoader().loadTestsFromTestCase(Tests) unittest.TextTestRunner(verbosity = 2).run(suite) if len(sys.argv) > 1 and sys.argv[1] == 'test': runTests(); else: reader = Reader(hasTestCount = False); solver = ProblemSolver(); printer = Printer(hasLineBetweenPrints = False); inputArray = reader.readFromConsole() while inputArray != []: solution = solver.run(inputArray) printer.printToConsole(solution) inputArray = reader.readFromConsole()
solution = [] start = input[0] end = input[1] for i in range(start, end + 1): if self.isPrime(i): solution.append(i); return solution # MAIN def runTests(): suite = unittest.TestLoader().loadTestsFromTestCase(Tests) unittest.TextTestRunner(verbosity = 2).run(suite) if len(sys.argv) > 1 and sys.argv[1] == 'test': runTests(); else: reader = Reader(hasTestCount = True); solver = ProblemSolver(); printer = Printer(hasLineBetweenPrints = True); inputArray = reader.readIntegersFromConsole() while inputArray != []: solution = solver.run(inputArray) printer.printToConsole(solution) inputArray = reader.readFromConsole()