#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import unittest
from singlecellmultiomics.utils import BlockZip

import os
"""
These tests check if the BlockZip module is working correctly
"""

class Test_BlockZip(unittest.TestCase):

    def test_read_write(self):
        zip_path = './data/test.bgzf'
        with BlockZip(zip_path,'w') as f:
            f.write('chr1',100,False,'yes')
            f.write('chr1',101,False,'yes')
            f.write('chr2',2101,True,'yes')
            f.write('chrX',0,False,'X')

        with BlockZip(zip_path,'r') as f:
            self.assertEqual( f[ ('chr1',100,False) ], 'yes' )
            self.assertEqual( f[ ('chr1',0,False) ], None)
            self.assertEqual( f[ ('chrX',0,False) ], 'X' )
            self.assertEqual( f[ ('chr2',2101,True) ], 'yes' )
            self.assertEqual( f[ ('chr1',0,True) ], None)


        # Test verification:
        with BlockZip(zip_path,'r') as f:
            f.verify()

        with BlockZip(zip_path,'w') as f:
            f.write('chr1',101,False,'yes')
            f.write('chr1',100,False,'yes')
            f.write('chr2',2101,True,'yes')
            f.write('chrX',0,False,'X')

        with BlockZip(zip_path,'r') as bz:
            self.assertRaises(ValueError, bz.verify)


        # Test for missing index:
        os.remove(zip_path+'.idx')
        self.assertRaises(ValueError, BlockZip, './non_existing.bgzf','r')

        os.remove(zip_path)
        self.assertRaises(ValueError, BlockZip, './non_existing.bgzf','r')




if __name__ == '__main__':
    unittest.main()