text: ###############################################################################
  #
  # Tests for XlsxWriter.
  #
  # Copyright (c), 2013-2018, John McNamara, jmcnamara@cpan.org
  #
  
  from ..excel_comparsion_test import ExcelComparisonTest
  from ...workbook import Workbook
  from ...compatibility import BytesIO
  
  
  class TestCompareXLSXFiles(ExcelComparisonTest):
      """
      Test file created by XlsxWriter against a file created by Excel.
  
      """
  
      def setUp(self):
  
          self.set_filename('header_image01.xlsx')
  
          self.ignore_elements = {'xl/worksheets/sheet1.xml': ['<pageMargins', '<pageSetup']}
  
      def test_create_file(self):
          """Test the creation of a simple XlsxWriter file with image(s)."""
  
          workbook = Workbook(self.got_filename)
  
          worksheet = workbook.add_worksheet()
  
          worksheet.set_header('&L&G',
                               {'image_left': self.image_dir + 'red.jpg'})
  
          workbook.close()
  
          self.assertExcelEqual()
  
      def test_create_file_in_memory(self):
          """Test the creation of a simple XlsxWriter file with image(s)."""
  
          workbook = Workbook(self.got_filename, {'in_memory': True})
  
          worksheet = workbook.add_worksheet()
  
          worksheet.set_header('&L&G',
                               {'image_left': self.image_dir + 'red.jpg'})
  
          workbook.close()
  
          self.assertExcelEqual()
  
      def test_create_file_from_bytesio(self):
          """Test the creation of a simple XlsxWriter file with image(s)."""
  
          workbook = Workbook(self.got_filename)
  
          worksheet = workbook.add_worksheet()
  
          image_file = open(self.image_dir + 'red.jpg', 'rb')
          image_data = BytesIO(image_file.read())
          image_file.close()
  
          worksheet.set_header('&L&G',
                               {'image_left': 'red.jpg',
                                'image_data_left': image_data})
  
          workbook.close()
  
          self.assertExcelEqual()
  
      def test_create_file_from_bytesio_in_memory(self):
          """Test the creation of a simple XlsxWriter file with image(s)."""
  
          workbook = Workbook(self.got_filename, {'in_memory': True})
  
          worksheet = workbook.add_worksheet()
  
          image_file = open(self.image_dir + 'red.jpg', 'rb')
          image_data = BytesIO(image_file.read())
          image_file.close()
  
          worksheet.set_header('&L&G',
                               {'image_left': 'red.jpg',
                                'image_data_left': image_data})
  
          workbook.close()
  
          self.assertExcelEqual()
  