added test cases for bzip compressed files

1 files changed, 18 insertions, 3 deletions

diff --git a/tests/test_trace.py b/tests/test_trace.py
index 24c96c5..65a9f5b 100644
--- a/tests/test_trace.py
+++ b/tests/test_trace.py
@@ -18,9 +18,26 @@ def test_load_gzip_compressed_v0dot1_trace():
     tr = loadTrace('tests/trace_test_data/xtrace1.dat.gz')
     assert isItExpectedTrace(tr) == True
 
+def test_load_bzip_compressed_v0dot1_trace():
+    tr = loadTrace('tests/trace_test_data/xtrace1.dat.bz')
+    assert isItExpectedTrace(tr) == True
+
+    tr = loadTrace('tests/trace_test_data/xtrace1.dat.bz2')
+    assert isItExpectedTrace(tr) == True
+
 def test_tryCompressedIfMissing():
     fname = 'tests/trace_test_data/only_compressed_file1.dat'
-    # first we check that the guess is working
+    # we check that the guess is working for gzip compressed file (.gz)
+    tr = loadTrace(fname, tryCompressedIfMissing=True)
+    assert isItExpectedTrace(tr) == True
+
+    fname = 'tests/trace_test_data/only_compressed_file2.dat'
+    # we check that the guess is working for bzip compressed file (.bz)
+    tr = loadTrace(fname, tryCompressedIfMissing=True)
+    assert isItExpectedTrace(tr) == True
+
+    fname = 'tests/trace_test_data/only_compressed_file3.dat'
+    # we check that the guess is working for bzip compressed file (.bz2)
     tr = loadTrace(fname, tryCompressedIfMissing=True)
     assert isItExpectedTrace(tr) == True
 
@@ -28,5 +45,3 @@ def test_tryCompressedIfMissing():
     with pytest.raises(FileNotFoundError) as exc_info:
         tr = loadTrace(fname, tryCompressedIfMissing=False)
 
-
-