code_ruby/test/maasha/test_bitarray.rb

   1 #!/usr/bin/env ruby
   2 $:.unshift File.join(File.dirname(__FILE__),'..','lib')
   3
   4 # Copyright (C) 2007-2010 Martin A. Hansen.
   5
   6 # This program is free software; you can redistribute it and/or
   7 # modify it under the terms of the GNU General Public License
   8 # as published by the Free Software Foundation; either version 2
   9 # of the License, or (at your option) any later version.
  10
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  19
  20 # http://www.gnu.org/copyleft/gpl.html
  21
  22 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  23
  24 # This software is part of the Biopieces framework (www.biopieces.org).
  25
  26 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  27
  28
  29 require 'maasha/bitarray'
  30 require 'test/unit'
  31 require 'pp'
  32
  33 class TestBitArray < Test::Unit::TestCase
  34   def setup
  35     @ba = BitArray.new(10)
  36   end
  37
  38   def test_BitArray_initialize_raises_on_bad_sizes
  39     [ -1, 0, 1.1, "a" ].each do |size|
  40       assert_raise(BitArrayError) { BitArray.new(size) }
  41     end
  42   end
  43
  44   def test_BitArray_initialize_dont_raise_on_ok_sizes
  45     [ 1, 10, 1000 ].each do |size|
  46       assert_nothing_raised { BitArray.new(size) }
  47     end
  48   end
  49
  50   def test_BitArray_size_returns_correctly
  51     assert_equal(10, @ba.size)
  52   end
  53
  54   def test_BitArray_to_s_returns_correctly
  55     assert_equal("0000000000", @ba.to_s)
  56   end
  57
  58   def test_BitArray_fill_returns_correctly
  59     assert_equal("1111111111", @ba.fill!.to_s)
  60   end
  61
  62   def test_BitArray_clear_returns_correctly
  63     assert_equal("0000000000", @ba.fill!.clear!.to_s)
  64   end
  65
  66   def test_BitArray_bit_set_with_bad_pos_raises
  67     [-1, 10, 1.1].each do |pos|
  68       assert_raise(BitArrayError) { @ba.bit_set(pos) }
  69     end
  70   end
  71
  72   def test_BitArray_bit_unset
  73     @ba.fill!
  74
  75     str = "1111111111"
  76
  77     (0.upto 9).each do |pos|
  78       @ba.bit_unset(pos)
  79       str[pos] = "0"
  80       assert_equal(str, @ba.to_s)
  81     end
  82   end
  83
  84   def test_BitArray_bit_unset_with_bad_pos_raises
  85     [-1, 10, 1.1].each do |pos|
  86       assert_raise(BitArrayError) { @ba.bit_unset(pos) }
  87     end
  88   end
  89
  90   def test_BitArray_bit_set_questionmark_with_bad_pos_raises
  91     [-1, 10, 1.1].each do |pos|
  92       assert_raise(BitArrayError) { @ba.bit_set?(pos) }
  93     end
  94   end
  95
  96   def test_BitArray_bit_set_questionmark
  97     (0.upto 9).each do |pos|
  98       @ba.bit_set(pos)
  99       assert_equal(true, @ba.bit_set?(pos))
 100     end
 101   end
 102
 103   def test_BitArray_bits_on_returns_correctly
 104     @ba.bit_set(4)
 105     @ba.bit_set(0)
 106     @ba.bit_set(1)
 107     assert_equal(3, @ba.bits_on)
 108   end
 109
 110   def test_BitArray_bits_off_returns_correctly
 111     @ba.bit_set(4)
 112     assert_equal(9, @ba.bits_off)
 113   end
 114
 115   def test_BitArray_AND_with_uneven_sizes_raises
 116     ba = BitArray.new(11)
 117     assert_raise(BitArrayError) { @ba & ba }
 118   end
 119
 120   def test_BitArray_AND_returns_correctly
 121     ba = BitArray.new(10)
 122     @ba.bit_set(4)
 123     @ba.bit_set(5)
 124     ba.bit_set(5)
 125     ba.bit_set(6)
 126     assert_equal("0000010000", (@ba & ba).to_s)
 127   end
 128
 129   def test_BitArray_OR_with_uneven_sizes_raises
 130     ba = BitArray.new(11)
 131     assert_raise(BitArrayError) { @ba | ba }
 132   end
 133
 134   def test_BitArray_OR_returns_correctly
 135     ba = BitArray.new(10)
 136     @ba.bit_set(4)
 137     @ba.bit_set(5)
 138     ba.bit_set(5)
 139     ba.bit_set(6)
 140     assert_equal("0000111000", (@ba | ba).to_s)
 141   end
 142
 143   def test_BitArray_XOR_with_uneven_sizes_raises
 144     ba = BitArray.new(11)
 145     assert_raise(BitArrayError) { @ba ^ ba }
 146   end
 147
 148   def test_BitArray_XOR_returns_correctly
 149     ba = BitArray.new(10)
 150     @ba.bit_set(4)
 151     @ba.bit_set(5)
 152     ba.bit_set(5)
 153     ba.bit_set(6)
 154     assert_equal("0000101000", (@ba ^ ba).to_s)
 155   end
 156
 157   def test_BitArray_FLIP_returns_correctly
 158     @ba.bit_set(0)
 159     @ba.bit_set(9)
 160
 161     assert_equal("0111111110", (~@ba).to_s)
 162   end
 163
 164   def test_interval_set_with_bad_interval_raises
 165     assert_raise(BitArrayError) { @ba.interval_set(-1, 4) }
 166     assert_raise(BitArrayError) { @ba.interval_set(1, 10) }
 167     assert_raise(BitArrayError) { @ba.interval_set(4, 2) }
 168   end
 169
 170   def test_interval_set_returns_correctly
 171     @ba.interval_set(0, 0)
 172     assert_equal("1000000000", @ba.to_s)
 173     @ba.interval_set(9, 9)
 174     assert_equal("1000000001", @ba.to_s)
 175     @ba.interval_set(2, 7)
 176     assert_equal("1011111101", @ba.to_s)
 177     @ba.interval_set(1, 8)
 178     assert_equal("1111111111", @ba.to_s)
 179   end
 180
 181   def test_interval_unset_with_bad_interval_raises
 182     assert_raise(BitArrayError) { @ba.interval_unset(-1, 4) }
 183     assert_raise(BitArrayError) { @ba.interval_unset(1, 10) }
 184     assert_raise(BitArrayError) { @ba.interval_unset(4, 2) }
 185   end
 186
 187   def test_interval_unset_returns_correctly
 188     ~@ba
 189
 190     @ba.interval_unset(0, 0)
 191     assert_equal("0111111111", @ba.to_s)
 192     @ba.interval_unset(9, 9)
 193     assert_equal("0111111110", @ba.to_s)
 194     @ba.interval_unset(2, 7)
 195     assert_equal("0100000010", @ba.to_s)
 196     @ba.interval_unset(1, 8)
 197     assert_equal("0000000000", @ba.to_s)
 198   end
 199
 200   def test_each_interval_returns_correctly
 201     @ba.fill!
 202     @ba.bit_unset(4)
 203
 204     assert_equal([[0, 3], [5, 9]], @ba.each_interval.to_a)
 205   end
 206 end
 207