1
0
Fork 0
mirror of https://github.com/ruby-opencv/ruby-opencv synced 2023-03-27 23:22:12 -04:00

Added a basic unit test.

This commit is contained in:
Francois Deschenes 2018-07-26 22:51:27 -07:00
parent 0dead60224
commit 061a5830e5
4 changed files with 2219 additions and 1 deletions

61
test/test_dnn.rb Normal file
View file

@ -0,0 +1,61 @@
#!/usr/bin/env ruby
# -*- mode: ruby; coding: utf-8 -*-
require 'opencv'
require File.expand_path(File.dirname(__FILE__)) + '/helper'
include Cv
class TestDnn < OpenCVTestCase
def test_read_net
c = Dnn.read_net(BVLC_GOOGLENET_PROTXT, BVLC_GOOGLENET_CAFFEMODEL)
assert_equal(Dnn::Net, c.class)
assert_raise(TypeError) {
Dnn.read_net(DUMMY_OBJ)
}
end
def test_read_net_from_caffe
c = Dnn.read_net_from_caffe(BVLC_GOOGLENET_PROTXT, BVLC_GOOGLENET_CAFFEMODEL)
assert_equal(Dnn::Net, c.class)
end
def test_blog_from_image
i = Cv.imread(FILENAME_GIRLS_PLAY_AND_PLANT_FLOWERS_IN_THE_PARK)
b = Dnn.blob_from_image(i, size: Cv::Size.new(224, 224))
assert_equal(Mat, b.class)
assert_equal(224, b.size(2))
assert_equal(224, b.size(3))
end
def test_net_initialize
c = Dnn::Net.new
assert_equal(Dnn::Net, c.class)
c = Dnn::Net.new(BVLC_GOOGLENET_PROTXT, BVLC_GOOGLENET_CAFFEMODEL)
assert_equal(Dnn::Net, c.class)
assert_raise(TypeError) {
Dnn::Net.new(DUMMY_OBJ)
}
end
def test_net_empty
c = Dnn::Net.new(BVLC_GOOGLENET_PROTXT, BVLC_GOOGLENET_CAFFEMODEL)
assert_equal(false, c.empty?)
c = Dnn::Net.new
assert_equal(true, c.empty?)
end
def test_net_forward
i = Cv.imread(FILENAME_GIRLS_PLAY_AND_PLANT_FLOWERS_IN_THE_PARK)
c = Dnn::Net.new(BVLC_GOOGLENET_PROTXT, BVLC_GOOGLENET_CAFFEMODEL)
c.input = Dnn.blob_from_image(i, size: Cv::Size.new(224, 224))
m = c.forward
assert_equal(Mat, m.class)
end
end