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")
