spec/02_slice_spec.rb in nmatrix-0.1.0.rc2 vs spec/02_slice_spec.rb in nmatrix-0.1.0.rc3
- old
+ new
@@ -312,9 +312,15 @@
context "operations" do
it "correctly transposes slices" do
expect(@m[0...3,0].transpose).to eq NMatrix[[0, 3, 6]]
+ expect(@m[0...3,1].transpose).to eq NMatrix[[1, 4, 7]]
+ expect(@m[0...3,2].transpose).to eq NMatrix[[2, 5, 8]]
+ expect(@m[0,0...3].transpose).to eq NMatrix[[0], [1], [2]]
+ expect(@m[1,0...3].transpose).to eq NMatrix[[3], [4], [5]]
+ expect(@m[2,0...3].transpose).to eq NMatrix[[6], [7], [8]]
+ expect(@m[1..2,1..2].transpose).to eq NMatrix[[4, 7], [5, 8]]
end
it "adds slices" do
expect(NMatrix[[0,0,0]] + @m[1,0..2]).to eq NMatrix[[3, 4, 5]]
end