import os
import unittest
from config import Config


class HelperMethodsTest(unittest.TestCase):

    def test_that_config_week_is_only_A_or_B(self):
        """
        This test ensures that the only value set for 'WEEK'
        in the envvars is A or B.
        """
        config = Config()

        if os.path.exists('./rtmbot.conf'):
            config.load_yaml('rtmbot.conf')

        else:
            config.load_os_environ_vars('FB__')

        week_options = ['A', 'B']
        self.assertIn(config['WEEK'], week_options)

import os
import unittest
from config import Config


class HelperMethodsTest(unittest.TestCase):

    def test_that_config_week_is_only_A_or_B(self):
        """
        This test ensures that the only value set for 'WEEK'
        in the envvars is A or B.
        """
        config = Config()
        week_options = ['A', 'B']

        if os.path.exists('./rtmbot.conf'):
            config.load_yaml('rtmbot.conf')
            self.assertIn(config['WEEK'], week_options)
        else:
            config.load_os_environ_vars('FB__')
            self.assertIn(config['FB__WEEK'], week_options)

Update tests for config week
Update tests for config week

  import os
  import unittest
  from config import Config
  
  
  class HelperMethodsTest(unittest.TestCase):
  
      def test_that_config_week_is_only_A_or_B(self):
          """
          This test ensures that the only value set for 'WEEK'
          in the envvars is A or B.
          """
          config = Config()
+         week_options = ['A', 'B']
  
          if os.path.exists('./rtmbot.conf'):
              config.load_yaml('rtmbot.conf')
- 
+             self.assertIn(config['WEEK'], week_options)
          else:
              config.load_os_environ_vars('FB__')
+             self.assertIn(config['FB__WEEK'], week_options)
  
-         week_options = ['A', 'B']
-         self.assertIn(config['WEEK'], week_options)
- 
Update tests for config week
## Code Before:
import os
import unittest
from config import Config


class HelperMethodsTest(unittest.TestCase):

    def test_that_config_week_is_only_A_or_B(self):
        """
        This test ensures that the only value set for 'WEEK'
        in the envvars is A or B.
        """
        config = Config()

        if os.path.exists('./rtmbot.conf'):
            config.load_yaml('rtmbot.conf')

        else:
            config.load_os_environ_vars('FB__')

        week_options = ['A', 'B']
        self.assertIn(config['WEEK'], week_options)

## Instruction:
Update tests for config week
## Code After:
import os
import unittest
from config import Config


class HelperMethodsTest(unittest.TestCase):

    def test_that_config_week_is_only_A_or_B(self):
        """
        This test ensures that the only value set for 'WEEK'
        in the envvars is A or B.
        """
        config = Config()
        week_options = ['A', 'B']

        if os.path.exists('./rtmbot.conf'):
            config.load_yaml('rtmbot.conf')
            self.assertIn(config['WEEK'], week_options)
        else:
            config.load_os_environ_vars('FB__')
            self.assertIn(config['FB__WEEK'], week_options)
