#123 Fix for the files with only one vertex

Author: Mikhail57

pywavefront/obj.py

@@ -139,6 +139,8 @@ def consume_vertices(self):
             try:
                 self.next_line()
             except StopIteration:
+                self.line = None
+                self.values = None
                 break
 
             if not self.values:

tests/fixtures/simple_one_vertex.obj

@@ -0,0 +1,2 @@
+o Cube
+v -0.500639 1.816088 1.058783

tests/test_parser.py

@@ -126,6 +126,14 @@ def testObjNoNormals(self):
 
         self.assertEqual(self.mesh2.materials[0].vertex_format, "T2F_V3F")
 
+    def testOnlyOneVertex(self):
+        meshes = pywavefront.Wavefront(fixture('simple_one_vertex.obj'))
+        self.vertices = meshes.vertices
+        self.meshes = meshes.mesh_list
+
+        self.assertEqual(1, len(self.meshes))
+        self.assertEqual(1, len(self.vertices))
+
     def testObjNoUVs(self):
         """Parse object with no uvs"""
         meshes = pywavefront.Wavefront(fixture('simple_normals.obj'))