def test_05_test_Visit_LogQueue(self): queue = Queue(6); for i in range(5): queue.enqueue(i); queue.accept(FilePrinter("./ThisIsATestFile.txt","")); with open('./ThisIsATestFile.txt', 'r') as myfile: result=myfile.read(); self.assertEqual(result, "\n|0|1|2|3|4|\n");
def testVisitLogQueue(self): queue = Queue(10) for i in range(5): queue.enqueue(i) queue.accept(ScreenPrinter("")) self.assertEqual(self.capturedOutput.getvalue(), "\n\n|0|1|2|3|4|\n\n")
def testVisitLogQueue(self): queue = Queue(10) for i in range(5): queue.enqueue(i) queue.accept(FilePrinter("./test.txt", "")) with open('./test.txt', 'r') as myfile: result = myfile.read() self.assertEqual(result, "\n|0|1|2|3|4|\n")
class FilePrinterTest(unittest.TestCase): def setUp(self): self.printer = FilePrinter("testFile.txt", "testName") self.linkedList = LinkedList() self.stack = Stack(3) self.queue = Queue(3) def tearDown(self): try: os.remove(self.printer.file_path) except IOError: pass # Test __init__() def test_a_Init(self): self.assertEqual(self.printer.file_path, "testFile.txt") self.assertEqual(self.printer.name, "testName") def test_b_VisitLinkedList(self): self.linkedList.append(1) self.linkedList.append(2) self.linkedList.append(3) self.linkedList.accept(self.printer) with open(self.printer.file_path, 'r') as testFile: self.assertEqual(testFile.read(), self.printer.name + "\n(1,2,3)\n") def test_c_VisitStack(self): self.stack.push(1) self.stack.push(2) self.stack.push(3) self.stack.accept(self.printer) filler="\n-------\n" with open(self.printer.file_path, 'r') as testFile: self.assertEqual(testFile.read(), self.printer.name +filler+' 3 '+filler+' 2 '+filler+' 1 '+filler) def test_d_VisitLogQueue(self): self.queue.enqueue(1) self.queue.enqueue(2) self.queue.enqueue(3) self.queue.accept(self.printer) with open(self.printer.file_path, 'r') as testFile: self.assertEqual(testFile.read(), self.printer.name + "\n|1|2|3|\n")
class ScreenPrinterTest(unittest.TestCase): def setUp(self): self.screenPrinter = ScreenPrinter("test") self.linkedList = LinkedList() self.stack = Stack(3) self.queue = Queue(3) # Test __init__() def test_a_Init(self): self.assertEqual(self.screenPrinter.name, "test") def test_b_VisitLinkedList(self): self.linkedList.append(1) self.linkedList.append(2) self.linkedList.append(3) sys.stdout = io.StringIO() self.linkedList.accept(ScreenPrinter("")) self.assertEqual(sys.stdout.getvalue(), "\n\n(1,2,3)\n\n") sys.stdout = sys.__stdout__ def test_c_VisitStack(self): self.stack.push(1) self.stack.push(2) self.stack.push(3) sys.stdout = io.StringIO() self.stack.accept(ScreenPrinter("")) filler = "\n-------\n" self.assertEqual( sys.stdout.getvalue(), '\n' + filler + ' 3 ' + filler + ' 2 ' + filler + ' 1 ' + filler + '\n') sys.stdout = sys.__stdout__ def test_d_VisitLogQueue(self): self.queue.enqueue(1) self.queue.enqueue(2) self.queue.enqueue(3) sys.stdout = io.StringIO() self.queue.accept(ScreenPrinter("")) self.assertEqual(sys.stdout.getvalue(), "\n\n|1|2|3|\n\n") sys.stdout = sys.__stdout__