from datetime import datetime, date, timedelta
import unittest

from businesstime.holidays.aus import QueenslandPublicHolidays, BrisbanePublicHolidays


class QueenslandPublicHolidaysTest(unittest.TestCase):

    def test_2016_08(self):
        holidays_gen = QueenslandPublicHolidays()
        self.assertEqual(
            list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
            []
        )


class BrisbanePublicHolidaysTest(unittest.TestCase):

    def test_2016_08(self):
        holidays_gen = BrisbanePublicHolidays()
        self.assertEqual(
            list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
            [
                date(2016, 8, 10)
            ]
        )

    def test_out_of_range(self):
        holidays_gen = BrisbanePublicHolidays()
        with self.assertRaises(NotImplementedError):
            list(holidays_gen(date(2017, 1, 1), end=date(2017, 12, 31)))

from datetime import datetime, date, timedelta
import unittest

from businesstime.holidays.aus import QueenslandPublicHolidays, BrisbanePublicHolidays


class QueenslandPublicHolidaysTest(unittest.TestCase):

    def test_2016_08(self):
        holidays_gen = QueenslandPublicHolidays()
        self.assertEqual(
            list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
            []
        )


class BrisbanePublicHolidaysTest(unittest.TestCase):

    def test_2016_08(self):
        holidays_gen = BrisbanePublicHolidays()
        self.assertEqual(
            list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
            [
                date(2016, 8, 10)
            ]
        )

    def test_out_of_range(self):
        holidays_gen = BrisbanePublicHolidays()
        def test():
            return list(holidays_gen(date(2017, 1, 1), end=date(2017, 12, 31)))
        self.assertRaises(NotImplementedError, test)

Fix tests in python 2.6
Fix tests in python 2.6

  from datetime import datetime, date, timedelta
  import unittest
  
  from businesstime.holidays.aus import QueenslandPublicHolidays, BrisbanePublicHolidays
  
  
  class QueenslandPublicHolidaysTest(unittest.TestCase):
  
      def test_2016_08(self):
          holidays_gen = QueenslandPublicHolidays()
          self.assertEqual(
              list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
              []
          )
  
  
  class BrisbanePublicHolidaysTest(unittest.TestCase):
  
      def test_2016_08(self):
          holidays_gen = BrisbanePublicHolidays()
          self.assertEqual(
              list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
              [
                  date(2016, 8, 10)
              ]
          )
  
      def test_out_of_range(self):
          holidays_gen = BrisbanePublicHolidays()
-         with self.assertRaises(NotImplementedError):
+         def test():
-             list(holidays_gen(date(2017, 1, 1), end=date(2017, 12, 31)))
+             return list(holidays_gen(date(2017, 1, 1), end=date(2017, 12, 31)))
+         self.assertRaises(NotImplementedError, test)
  
Fix tests in python 2.6
## Code Before:
from datetime import datetime, date, timedelta
import unittest

from businesstime.holidays.aus import QueenslandPublicHolidays, BrisbanePublicHolidays


class QueenslandPublicHolidaysTest(unittest.TestCase):

    def test_2016_08(self):
        holidays_gen = QueenslandPublicHolidays()
        self.assertEqual(
            list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
            []
        )


class BrisbanePublicHolidaysTest(unittest.TestCase):

    def test_2016_08(self):
        holidays_gen = BrisbanePublicHolidays()
        self.assertEqual(
            list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
            [
                date(2016, 8, 10)
            ]
        )

    def test_out_of_range(self):
        holidays_gen = BrisbanePublicHolidays()
        with self.assertRaises(NotImplementedError):
            list(holidays_gen(date(2017, 1, 1), end=date(2017, 12, 31)))

## Instruction:
Fix tests in python 2.6
## Code After:
from datetime import datetime, date, timedelta
import unittest

from businesstime.holidays.aus import QueenslandPublicHolidays, BrisbanePublicHolidays


class QueenslandPublicHolidaysTest(unittest.TestCase):

    def test_2016_08(self):
        holidays_gen = QueenslandPublicHolidays()
        self.assertEqual(
            list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
            []
        )


class BrisbanePublicHolidaysTest(unittest.TestCase):

    def test_2016_08(self):
        holidays_gen = BrisbanePublicHolidays()
        self.assertEqual(
            list(holidays_gen(date(2016, 8, 1), end=date(2016, 8, 31))),
            [
                date(2016, 8, 10)
            ]
        )

    def test_out_of_range(self):
        holidays_gen = BrisbanePublicHolidays()
        def test():
            return list(holidays_gen(date(2017, 1, 1), end=date(2017, 12, 31)))
        self.assertRaises(NotImplementedError, test)
