示例#1
0
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
示例#2
0
    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))
示例#3
0
    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)
示例#4
0
    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)
示例#5
0
 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)
示例#6
0
 def test_hashable(self):
     s = set([FileReader("foo")])
     s.add(FileReader("foo"))
     self.assertEquals(1, len(s))
示例#7
0
 def test_equality(self):
     self.assertEquals(FileReader("foo"), FileReader("foo"))
     self.assertNotEquals(FileReader("foo"), FileReader("bar"))
     self.assertNotEquals(FileReader("foo"), 1)
示例#8
0
 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)