def test_find_first_heading_no_heading(self): stream = StringIO.StringIO("#Start-message\n#Id=PD-gs2exp\n") line = gs2._find_first_heading(stream) self.assertIs(line, None)
def test_find_first_heading_empty_file(self): stream = StringIO.StringIO("") line = gs2._find_first_heading(stream) self.assertIs(line, None)
def test_find_first_heading(self): stream = StringIO.StringIO("##Start-message\n#Id=PD-gs2exp\n") line = gs2._find_first_heading(stream) self.assertEqual(line, "##Start-message\n")