def _build_file_readers(input_dir): in_files = glob.glob(os.path.join(input_dir, "*")) file_readers = [] for in_file in in_files: file_readers.append(FileReader(in_file)) return file_readers
def testCompare(self): expected_readers = [ FileReader("1A.txt"), FileReader("1B.txt"), FileReader("2A.txt"), FileReader("10A.txt"), FileReader("10B.txt"), FileReader("11A.txt"), FileReader("11B.txt"), FileReader("20A.txt"), FileReader("100A.txt") ] input_readers = expected_readers[::-1] self.assertEquals(expected_readers, sorted(input_readers))
def test_read_lines(self): with TempDirectory() as input_file: input_file.write("A.tmp", b"1\n2\n3") reader = FileReader(os.path.join(input_file.path, "A.tmp")) reader.open() actual_lines = [line for line in reader.read_lines()] reader.close() self.assertEquals(["1\n", "2\n", "3"], actual_lines)
def test_read_lines_raisesTypeErrorWhenClosed(self): with TempDirectory() as input_file: input_file.write("A.tmp", b"1\n2\n3") reader = FileReader(os.path.join(input_file.path, "A.tmp")) line_iter = reader.read_lines() self.assertRaises(TypeError, next, line_iter)
def test_hashable(self): s = set([FileReader("foo")]) s.add(FileReader("foo")) self.assertEquals(1, len(s))
def test_equality(self): self.assertEquals(FileReader("foo"), FileReader("foo")) self.assertNotEquals(FileReader("foo"), FileReader("bar")) self.assertNotEquals(FileReader("foo"), 1)