numo-narray/spec/bit_spec.rb at e32658555a941c58e713e8066c415aa0872ccae2 · ruby-numo/numo-narray · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
require_relative '../lib/numo/narray.so'
#Numo::NArray.debug = true

RSpec.configure do |config|
  config.filter_run :focus
  config.run_all_when_everything_filtered = true
end
#context :focus=>true do ... end

dtype = Numo::Bit

describe dtype do
  it{expect(dtype).to be < Numo::NArray}
end

procs = [
  [proc{|tp,a| tp[*a] },""],
  [proc{|tp,a| tp[*a][true] },"[true]"],
  [proc{|tp,a| tp[*a][0..-1] },"[0..-1]"]
]
procs.each do |init,ref|

  describe dtype,"[0,1,1,0,1,0,0,1]"+ref do
    before(:all) do
      @src = [0,1,1,0,1,0,0,1]
      @n = @src.size
      @a = init.call(dtype,@src)
    end

    it{expect(@a).to eq @src}
    it{expect(@a & 0).to eq [0]*@n}
    it{expect(@a & 1).to eq @src}
    it{expect(@a | 0).to eq @src}
    it{expect(@a | 1).to eq [1]*@n}
    it{expect(@a ^ 0).to eq @src.map{|x| x^0}}
    it{expect(@a ^ 1).to eq @src.map{|x| x^1}}
    it{expect(~@a).to eq @src.map{|x| 1-x}}

    it{expect(@a.count_true).to eq 4}
    it{expect(@a.count_false).to eq 4}
    it{expect(@a.where).to eq [1,2,4,7]}
    it{expect(@a.where2).to eq [[1,2,4,7],[0,3,5,6]]}
    it{expect(@a.mask(Numo::DFloat[1,2,3,4,5,6,7,8])).to eq [2,3,5,8]}
    it{expect(@a).not_to be_all}
    it{expect(@a).to     be_any}
    it{expect(@a).not_to be_none}

    after(:all) do
      @a = nil
    end
  end

end

procs = [
  [proc{|tp,a| tp[*a] },""],
  [proc{|tp,a| tp[*a][true,0..-1] },"[true,true]"],
]
procs.each do |init,ref|

  describe dtype,"[[0,1,1,0],[1,0,0,1]]"+ref do
    before(:all) do
      @src = [[0,1,1,0],[1,0,0,1]]
      @n = @src.size
      @a = init.call(dtype,@src)
    end

    it{expect(@a[5]).to eq 0}
    it{expect(@a[-1]).to eq 1}
    it{expect(@a[1,0]).to eq @src[1][0]}
    it{expect(@a[1,1]).to eq @src[1][1]}
    it{expect(@a[1,2]).to eq @src[1][2]}
    it{expect(@a[3..4]).to eq [0,1]}
    it{expect(@a[0,1..2]).to eq [1,1]}
    it{expect(@a[0,:*]).to eq @src[0]}
    it{expect(@a[1,:*]).to eq @src[1]}
    it{expect(@a[:*,1]).to eq [@src[0][1],@src[1][1]]}

    it{expect(@a.count_true).to eq 4}
    it{expect(@a.count_false).to eq 4}
    it{expect(@a.where).to eq [1,2,4,7]}
    it{expect(@a.where2).to eq [[1,2,4,7],[0,3,5,6]]}
    it{expect(@a.mask(Numo::DFloat[[1,2,3,4],[5,6,7,8]])).to eq [2,3,5,8]}
    it{expect(@a).not_to be_all}
    it{expect(@a).to     be_any}
    it{expect(@a).not_to be_none}

    after(:all) do
      @a = nil
    end
  end

end