from unittest.mock import Mock, patch
from django.test import TestCase
from data_refinery_common import microarray

CEL_FILE_HUMAN = "test-files/C30057.CEL"
CEL_FILE_RAT = "test-files/SG2_u34a.CEL"
CEL_FILE_MOUSE = "test-files/97_(Mouse430_2).CEL"
CEL_FILE_ZEBRAFISH = "test-files/CONTROL6.cel"


class MicroarrayTestCase(TestCase):

    def test_get_platform_from_CEL(self):
        self.assertEqual("hgu95av2", microarray.get_platform_from_CEL(CEL_FILE_HUMAN))
        self.assertEqual("rgu34a", microarray.get_platform_from_CEL(CEL_FILE_RAT))
        self.assertEqual("mouse4302", microarray.get_platform_from_CEL(CEL_FILE_MOUSE))
        self.assertEqual("zebgene11st", microarray.get_platform_from_CEL(CEL_FILE_ZEBRAFISH))

from unittest.mock import Mock, patch
from django.test import TestCase
from data_refinery_common import microarray

CEL_FILE_HUMAN = "test-files/C30057.CEL.gz"
CEL_FILE_RAT = "test-files/SG2_u34a.CEL.gz"
CEL_FILE_MOUSE = "test-files/97_(Mouse430_2).CEL.gz"
CEL_FILE_ZEBRAFISH = "test-files/CONTROL6.cel.gz"


class MicroarrayTestCase(TestCase):

    def test_get_platform_from_CEL(self):
        self.assertEqual("hgu95av2", microarray.get_platform_from_CEL(CEL_FILE_HUMAN))
        self.assertEqual("rgu34a", microarray.get_platform_from_CEL(CEL_FILE_RAT))
        self.assertEqual("mouse4302", microarray.get_platform_from_CEL(CEL_FILE_MOUSE))
        self.assertEqual("zebgene11st", microarray.get_platform_from_CEL(CEL_FILE_ZEBRAFISH))

Update test file paths for common to point to compressed versions.
Update test file paths for common to point to compressed versions.

  from unittest.mock import Mock, patch
  from django.test import TestCase
  from data_refinery_common import microarray
  
- CEL_FILE_HUMAN = "test-files/C30057.CEL"
+ CEL_FILE_HUMAN = "test-files/C30057.CEL.gz"
- CEL_FILE_RAT = "test-files/SG2_u34a.CEL"
+ CEL_FILE_RAT = "test-files/SG2_u34a.CEL.gz"
- CEL_FILE_MOUSE = "test-files/97_(Mouse430_2).CEL"
+ CEL_FILE_MOUSE = "test-files/97_(Mouse430_2).CEL.gz"
- CEL_FILE_ZEBRAFISH = "test-files/CONTROL6.cel"
+ CEL_FILE_ZEBRAFISH = "test-files/CONTROL6.cel.gz"
  
  
  class MicroarrayTestCase(TestCase):
  
      def test_get_platform_from_CEL(self):
          self.assertEqual("hgu95av2", microarray.get_platform_from_CEL(CEL_FILE_HUMAN))
          self.assertEqual("rgu34a", microarray.get_platform_from_CEL(CEL_FILE_RAT))
          self.assertEqual("mouse4302", microarray.get_platform_from_CEL(CEL_FILE_MOUSE))
          self.assertEqual("zebgene11st", microarray.get_platform_from_CEL(CEL_FILE_ZEBRAFISH))
  
Update test file paths for common to point to compressed versions.
## Code Before:
from unittest.mock import Mock, patch
from django.test import TestCase
from data_refinery_common import microarray

CEL_FILE_HUMAN = "test-files/C30057.CEL"
CEL_FILE_RAT = "test-files/SG2_u34a.CEL"
CEL_FILE_MOUSE = "test-files/97_(Mouse430_2).CEL"
CEL_FILE_ZEBRAFISH = "test-files/CONTROL6.cel"


class MicroarrayTestCase(TestCase):

    def test_get_platform_from_CEL(self):
        self.assertEqual("hgu95av2", microarray.get_platform_from_CEL(CEL_FILE_HUMAN))
        self.assertEqual("rgu34a", microarray.get_platform_from_CEL(CEL_FILE_RAT))
        self.assertEqual("mouse4302", microarray.get_platform_from_CEL(CEL_FILE_MOUSE))
        self.assertEqual("zebgene11st", microarray.get_platform_from_CEL(CEL_FILE_ZEBRAFISH))

## Instruction:
Update test file paths for common to point to compressed versions.
## Code After:
from unittest.mock import Mock, patch
from django.test import TestCase
from data_refinery_common import microarray

CEL_FILE_HUMAN = "test-files/C30057.CEL.gz"
CEL_FILE_RAT = "test-files/SG2_u34a.CEL.gz"
CEL_FILE_MOUSE = "test-files/97_(Mouse430_2).CEL.gz"
CEL_FILE_ZEBRAFISH = "test-files/CONTROL6.cel.gz"


class MicroarrayTestCase(TestCase):

    def test_get_platform_from_CEL(self):
        self.assertEqual("hgu95av2", microarray.get_platform_from_CEL(CEL_FILE_HUMAN))
        self.assertEqual("rgu34a", microarray.get_platform_from_CEL(CEL_FILE_RAT))
        self.assertEqual("mouse4302", microarray.get_platform_from_CEL(CEL_FILE_MOUSE))
        self.assertEqual("zebgene11st", microarray.get_platform_from_CEL(CEL_FILE_ZEBRAFISH))
