One of the AlgorithmQuiz

See Seminar:CountingCodeLines

{{{#!python
import unittest
from cStringIO import StringIO

def countLines(aFile):
    nlines = 0
    inComment = False
    for line in aFile:
        line = line.strip()
        if line.startswith('//'):
            continue
        if line.startswith('/*'):
            inComment = False
        if line.find('*/') >= 0:
            inComment = True
        if line and not inComment:
            nlines+=1
    return nlines

class Test(unittest.TestCase):
    def test1(self):
        input = StringIO("""\
// This file contains 3 lines of code
public interface Dave {
  /**
   * count the number of lines in a file
   */
   int countLines(File inFile); // not the real signature!
}""")
        self.assertEquals(3, countLines(input))

    def test2(self):
        input = StringIO("""\
/*****
* This is a test program with 5 lines of code
*  \/* no nesting allowed!
//*****//***/// Slightly pathological comment ending...

public class Hello {
    public static final void main(String [] args) { // gotta love Java
      // Say hello
      System./*wait*/out./*for*/println/*it*/("Hello/*");
   }

}""")
        self.assertEquals(5, countLines(input))

if __name__=='__main__':
    unittest.main()
}}}