require_relative 'main'
require 'test/unit'

class TestNumSpecial < Test::Unit::TestCase
  def test_simple
    assert_equal(1, num_special([[1, 0, 0], [0, 0, 1], [1, 0, 0]]))
  end

  def test_multiple
    assert_equal(3, num_special([[1, 0, 0], [0, 1, 0], [0, 0, 1]]))
  end
end