import pypolycomp
import numpy as np

def test_compression():
    for cur_type in (np.int8, np.int16, np.int32, np.int64,
                     np.uint8, np.uint16, np.uint32, np.uint64):
        compressed = pypolycomp.rle_compress(np.array([1, 1, 1, 2, 3], dtype=cur_type))
        assert np.all(compressed == np.array([3, 1, 1, 2, 1, 3], dtype=cur_type))


import pypolycomp
import numpy as np

def test_compression():
    for cur_type in (np.int8, np.int16, np.int32, np.int64,
                     np.uint8, np.uint16, np.uint32, np.uint64):
        compressed = pypolycomp.rle_compress(np.array([1, 1, 1, 2, 3], dtype=cur_type))
        assert np.all(compressed == np.array([3, 1, 1, 2, 1, 3], dtype=cur_type))

def test_decompression():
    for cur_type in (np.int8, np.int16, np.int32, np.int64,
                     np.uint8, np.uint16, np.uint32, np.uint64):
        input_values = np.array(np.random.randint(100, size=1000),
                                dtype=cur_type)
        compressed = pypolycomp.rle_compress(input_values)
        output_values = pypolycomp.rle_decompress(compressed)
        assert np.all(input_values == output_values)

Add test for RLE decompression
Add test for RLE decompression

  import pypolycomp
  import numpy as np
  
  def test_compression():
      for cur_type in (np.int8, np.int16, np.int32, np.int64,
                       np.uint8, np.uint16, np.uint32, np.uint64):
          compressed = pypolycomp.rle_compress(np.array([1, 1, 1, 2, 3], dtype=cur_type))
          assert np.all(compressed == np.array([3, 1, 1, 2, 1, 3], dtype=cur_type))
  
+ def test_decompression():
+     for cur_type in (np.int8, np.int16, np.int32, np.int64,
+                      np.uint8, np.uint16, np.uint32, np.uint64):
+         input_values = np.array(np.random.randint(100, size=1000),
+                                 dtype=cur_type)
+         compressed = pypolycomp.rle_compress(input_values)
+         output_values = pypolycomp.rle_decompress(compressed)
+         assert np.all(input_values == output_values)
  
Add test for RLE decompression
## Code Before:
import pypolycomp
import numpy as np

def test_compression():
    for cur_type in (np.int8, np.int16, np.int32, np.int64,
                     np.uint8, np.uint16, np.uint32, np.uint64):
        compressed = pypolycomp.rle_compress(np.array([1, 1, 1, 2, 3], dtype=cur_type))
        assert np.all(compressed == np.array([3, 1, 1, 2, 1, 3], dtype=cur_type))


## Instruction:
Add test for RLE decompression
## Code After:
import pypolycomp
import numpy as np

def test_compression():
    for cur_type in (np.int8, np.int16, np.int32, np.int64,
                     np.uint8, np.uint16, np.uint32, np.uint64):
        compressed = pypolycomp.rle_compress(np.array([1, 1, 1, 2, 3], dtype=cur_type))
        assert np.all(compressed == np.array([3, 1, 1, 2, 1, 3], dtype=cur_type))

def test_decompression():
    for cur_type in (np.int8, np.int16, np.int32, np.int64,
                     np.uint8, np.uint16, np.uint32, np.uint64):
        input_values = np.array(np.random.randint(100, size=1000),
                                dtype=cur_type)
        compressed = pypolycomp.rle_compress(input_values)
        output_values = pypolycomp.rle_decompress(compressed)
        assert np.all(input_values == output_values)
