1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138
|
# Copyright (c) 2008 by Intevation GmbH
# Authors:
# Anthony Lenton <anthony@except.com.ar>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with Thuban for details.
"""
Tests for the PFM importer.
"""
import unittest
from StringIO import StringIO
from Extensions.importMP.pfm import MPParser, MPParseError
import snippets
class TestPFM (unittest.TestCase):
def testInitialValue(self):
""" Check that the parser has good initial values """
parser = MPParser()
self.assertEquals (parser.mapName, "")
self.assertEquals (parser.parsedLines, 0)
self.assertEquals (parser.parsedSections, 0)
self.assertEquals (parser.pois, [])
self.assertEquals (parser.polygons, [])
self.assertEquals (parser.polylines, [])
def testParsedLines(self):
""" Check that comments and blank lines are skipped correctly """
parser = MPParser()
parser.parse (StringIO(snippets.justHeader))
self.assertEquals (parser.parsedLines, 4)
parser = MPParser()
parser.parse (StringIO(snippets.headerWithComments))
self.assertEquals (parser.parsedLines, 12)
def testSections (self):
""" Check that sections are matched up properly """
parser = MPParser()
parser.parse (StringIO(snippets.justHeader))
self.assertEquals (parser.parsedSections, 1)
parser = MPParser()
mp = StringIO (snippets.nonsenseBounder)
self.assertRaises (MPParseError, parser.parse, mp)
parser = MPParser()
mp = StringIO (snippets.bounderMismatch)
self.assertRaises (MPParseError, parser.parse, mp)
parser = MPParser()
mp = StringIO (snippets.doubleSectionClose)
self.assertRaises (MPParseError, parser.parse, mp)
def testCaseInsensitive (self):
""" Check that no exception is raised if capitals are used"""
parser = MPParser()
parser.parse (StringIO (snippets.caseSensitivity1))
parser = MPParser()
parser.parse (StringIO (snippets.caseSensitivity2))
parser = MPParser()
parser.parse (StringIO (snippets.caseSensitivity3))
def testIncompleteSection (self):
""" Check that an exception is raised if the last section is
left open """
parser = MPParser()
mp = StringIO(snippets.incompleteSection)
self.assertRaises (MPParseError, parser.parse, mp)
def testHeaderSection (self):
""" Check that the header section is parsed """
parser = MPParser ()
mp = StringIO(snippets.nothingness)
self.assertRaises (MPParseError, parser.parse, mp)
parser = MPParser()
mp = StringIO(snippets.twoHeaderSections)
self.assertRaises (MPParseError, parser.parse, mp)
parser = MPParser()
parser.parse(StringIO(snippets.mapWithName))
self.assertEquals (parser.mapName, "My Test Map")
def testSample (self):
parser = MPParser()
parser.parse (StringIO(snippets.miniLobos))
self.assertEquals(len(parser.pois), 4)
self.assertEquals(len(parser.polylines), 2)
self.assertEquals(len(parser.polygons), 1)
def testPolylines (self):
parser = MPParser()
parser.parse (StringIO(snippets.RGN40s))
self.assertEquals(len(parser.polylines), 4)
parser = MPParser()
parser.parse (StringIO(snippets.mixedRGN40sAndPolylines))
self.assertEquals(len(parser.polylines), 4)
def testPolygons (self):
parser = MPParser()
parser.parse (StringIO(snippets.RGN80s))
self.assertEquals(len(parser.polygons), 4)
parser = MPParser()
parser.parse (StringIO(snippets.mixedRGN80sAndPolygons))
self.assertEquals(len(parser.polygons), 4)
def testPois (self):
parser = MPParser()
parser.parse (StringIO(snippets.RGN10s))
self.assertEquals(len(parser.pois), 4)
parser = MPParser()
parser.parse (StringIO(snippets.RGN20s))
self.assertEquals(len(parser.pois), 4)
parser = MPParser()
parser.parse (StringIO(snippets.mixedPois))
self.assertEquals(len(parser.pois), 6)
def testDontCrashWithWeirdTypes (self):
parser = MPParser()
parser.parse (StringIO(snippets.weirdTypes))
self.assertEquals(len(parser.pois), 3)
self.assertEquals(len(parser.polylines), 2)
self.assertEquals(len(parser.polygons), 1)
def testMissingTwoCharFormatWorks (self):
""" Test that if the type 0xe works as well as 0x0e, for example"""
parser = MPParser()
parser.parse (StringIO(snippets.missingZeroType))
self.assertEquals(len(parser.polygons), 2)
self.assertEquals(parser.polygons[0]['parsedType'], "Airport Runway")
self.assertEquals(parser.polygons[1]['parsedType'], "Airport Runway")
def testTypesAreTheRightKind (self):
""" Test that types from polylines and polygons come from the
right list of types """
p = MPParser()
p.parse (StringIO(snippets.polyTypes))
self.assertEquals(p.polygons[0]['parsedType'], "City")
self.assertEquals(p.polylines[0]['parsedType'], "Major Highway-thick")
if __name__ == "__main__":
unittest.main()
|