94 lines
2.1 KiB
Ruby
Executable File
94 lines
2.1 KiB
Ruby
Executable File
#!/usr/bin/env ruby
|
|
# -*- mode: ruby; coding: utf-8-unix -*-
|
|
require 'test/unit'
|
|
require 'opencv'
|
|
require 'date'
|
|
require File.expand_path(File.dirname(__FILE__)) + '/helper'
|
|
|
|
include OpenCV
|
|
|
|
# Tests for OpenCV::EigenFaces
|
|
class TestEigenFaces < OpenCVTestCase
|
|
def setup
|
|
@eigenfaces = EigenFaces.new
|
|
|
|
@eigenfaces_trained = EigenFaces.new
|
|
@images = [CvMat.load(FILENAME_LENA256x256, CV_LOAD_IMAGE_GRAYSCALE)] * 2
|
|
@labels = [1, 2]
|
|
@eigenfaces_trained.train(@images, @labels)
|
|
end
|
|
|
|
def test_initialize
|
|
[EigenFaces.new, EigenFaces.new(1), EigenFaces.new(1, 99999)].each { |ef|
|
|
assert_equal(EigenFaces, ef.class)
|
|
}
|
|
|
|
assert_raise(TypeError) {
|
|
EigenFaces.new(DUMMY_OBJ)
|
|
}
|
|
|
|
assert_raise(TypeError) {
|
|
EigenFaces.new(1, DUMMY_OBJ)
|
|
}
|
|
end
|
|
|
|
def test_train
|
|
assert_nil(@eigenfaces.train(@images, @labels))
|
|
|
|
assert_raise(TypeError) {
|
|
@eigenfaces.train(DUMMY_OBJ, @labels)
|
|
}
|
|
|
|
assert_raise(TypeError) {
|
|
@eigenfaces.train(@images, DUMMY_OBJ)
|
|
}
|
|
end
|
|
|
|
def test_predict
|
|
predicted_label, predicted_confidence = @eigenfaces_trained.predict(@images[0])
|
|
assert_equal(@labels[0], predicted_label)
|
|
assert_in_delta(0.0, predicted_confidence, 0.01)
|
|
|
|
assert_raise(TypeError) {
|
|
@eigenfaces_trained.predict(DUMMY_OBJ)
|
|
}
|
|
end
|
|
|
|
def test_save
|
|
filename = "eigenfaces_save-#{DateTime.now.strftime('%Y%m%d%H%M%S')}.xml"
|
|
begin
|
|
@eigenfaces_trained.save(filename)
|
|
assert(File.exist? filename)
|
|
ensure
|
|
File.delete filename
|
|
end
|
|
assert_raise(TypeError) {
|
|
@eigenfaces_trained.save(DUMMY_OBJ)
|
|
}
|
|
end
|
|
|
|
def test_load
|
|
assert_nothing_raised {
|
|
@eigenfaces_trained.load('eigenfaces_save.xml')
|
|
}
|
|
assert_raise(TypeError) {
|
|
@eigenfaces_trained.load(DUMMY_OBJ)
|
|
}
|
|
end
|
|
|
|
def test_name
|
|
assert_equal('FaceRecognizer.Eigenfaces', @eigenfaces.name)
|
|
end
|
|
|
|
def test_get_mat
|
|
mat = @eigenfaces_trained.get_mat('eigenvalues')
|
|
assert_not_nil(mat)
|
|
assert_equal(CvMat, mat.class)
|
|
|
|
assert_raise(TypeError) {
|
|
@eigenfaces_trained.get_mat(DUMMY_OBJ)
|
|
}
|
|
end
|
|
end
|
|
|