from template import *

import os.path

class aubiopitch_test_case(program_test_case):

  import os.path
  filename = os.path.join('..','..','sounds','woodblock.aiff')
  progname = "PYTHONPATH=../../python:../../python/aubio/.libs " + \
              os.path.join('..','..','python','aubiopitch')

  def test_aubiopitch(self):
    """ test aubiopitch with default parameters """
    self.getOutput()
    # FIXME: useless check
    self.assertEqual(len(self.output.split('\n')), 1)
    #self.assertEqual(float(self.output.strip()), 0.)

  def test_aubiopitch_verbose(self):
    """ test aubiopitch with -v parameter """
    self.command += " -v "
    self.getOutput()
    # FIXME: loose checking: make sure at least 8 lines are printed
    assert len(self.output) >= 8

  def test_aubiopitch_devnull(self):
    """ test aubiopitch on /dev/null """
    self.filename = "/dev/null"
    # exit status should not be 0
    self.getOutput(expected_status = 256)
    # and there should be an error message
    assert len(self.output) > 0
    # that looks like this 
    output_lines = self.output.split('\n')
    #assert output_lines[0] == "Unable to open input file /dev/null."
    #assert output_lines[1] == "Supported file format but file is malformed."
    #assert output_lines[2] == "Could not open input file /dev/null."

mode_names = ["yinfft", "yin", "fcomb", "mcomb", "schmitt"]
for name in mode_names:
  exec("class aubiopitch_test_case_" + name + "(aubiopitch_test_case):\n\
    options = \" -m " + name + " \"")

class aubiopitch_test_yinfft(program_test_case):

  filename = os.path.join('..','..','sounds','16568_acclivity_TwoCows.wav')
  progname = "PYTHONPATH=../../python:../../python/aubio/.libs " + \
              os.path.join('..','..','python','aubiopitch')
  options  = " -m yinfft -t 0.75 "

  def test_aubiopitch(self):
    """ test aubiopitch with default parameters """
    self.getOutput()
    expected_output = open(os.path.join('examples','aubiopitch','yinfft'+'.'+os.path.basename(self.filename)+'.txt')).read()
    for line_out, line_exp in zip(self.output.split('\n'), expected_output.split('\n')):
      assert line_out == line_exp, line_exp + " vs. " + line_out

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