ruby-opencv/test/test_cvvideowriter.rb

#!/usr/bin/env ruby
# -*- mode: ruby; coding: utf-8-unix -*- 
require 'test/unit'
require 'opencv'
require File.expand_path(File.dirname(__FILE__)) + '/helper'

include OpenCV

# Tests for OpenCV::CvVideoWriter
class TestCvVideoWriter < OpenCVTestCase
  OUTPUT_FILENAME = 'videowriter_result.avi'
  
  def test_initialize
    vw = CvVideoWriter.new(OUTPUT_FILENAME, 'MJPG', 15, CvSize.new(320, 240))
    assert_equal(CvVideoWriter, vw.class)
    vw.close

    vw = CvVideoWriter.new(OUTPUT_FILENAME, 'MJPG', 15, CvSize.new(320, 240), false)
    assert_equal(CvVideoWriter, vw.class)
    vw.close

    ## Supported only Windows(?)
    # vw = CvVideoWriter.new(OUTPUT_FILENAME, nil, 15, CvSize.new(320, 240), false)
    # assert_equal(CvVideoWriter, vw.class)
    # vw.close

    CvVideoWriter.new(OUTPUT_FILENAME, 'MJPG', 15, CvSize.new(320, 240)) { |vw|
      assert_equal(CvVideoWriter, vw.class)      
    }

    assert_raise(TypeError) {
      vw = CvVideoWriter.new(123, 'MJPG', 15, CvSize.new(320, 240), false)
    }
  end

  def test_write
    img = IplImage.load(FILENAME_LENA256x256)
    vw = CvVideoWriter.new(OUTPUT_FILENAME, 'MJPG', 15, CvSize.new(256, 256))
    vw.write img
    vw.close

    CvVideoWriter.new(OUTPUT_FILENAME, 'MJPG', 15, CvSize.new(256, 256)) { |vw|
      vw.write img
    }
    
    assert_raise(TypeError) {
      CvVideoWriter.new(OUTPUT_FILENAME, 'MJPG', 15, CvSize.new(256, 256)) { |vw|
        vw.write DUMMY_OBJ
      }
    }
  end

  def test_close
    vw = CvVideoWriter.new(OUTPUT_FILENAME, 'MJPG', 15, CvSize.new(320, 240))
    vw.close
  end
end