<s> from __future__ import print_function <EOL> import os <EOL> import unittest <EOL> from airflow import configuration <EOL> from airflow . configuration import conf <EOL> configuration . test_mode ( ) <EOL> class ConfTest ( unittest . TestCase ) : <EOL> def setup ( self ) : <EOL> configuration . test_mode ( ) <EOL> def test_env_var_config ( self ) : <EOL> opt = conf . get ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> self . assertEqual ( opt , '<STR_LIT>' ) <EOL> def test_conf_as_dict ( self ) : <EOL> cfg_dict = conf . as_dict ( ) <EOL> self . assertEqual ( cfg_dict [ '<STR_LIT>' ] [ '<STR_LIT>' ] , '<STR_LIT:True>' ) <EOL> self . assertEqual ( cfg_dict [ '<STR_LIT>' ] [ '<STR_LIT>' ] , '<STR_LIT>' ) <EOL> conf . remove_option ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cfg_dict = conf . as_dict ( ) <EOL> self . assertEqual ( cfg_dict [ '<STR_LIT>' ] [ '<STR_LIT>' ] , '<STR_LIT:True>' ) <EOL> cfg_dict = conf . as_dict ( display_source = True ) <EOL> self . assertEqual ( cfg_dict [ '<STR_LIT>' ] [ '<STR_LIT>' ] [ <NUM_LIT:1> ] , '<STR_LIT>' ) <EOL> self . assertEqual ( cfg_dict [ '<STR_LIT>' ] [ '<STR_LIT>' ] [ <NUM_LIT:1> ] , '<STR_LIT:default>' ) <EOL> self . assertEqual ( <EOL> cfg_dict [ '<STR_LIT>' ] [ '<STR_LIT>' ] , ( '<STR_LIT>' , '<STR_LIT>' ) ) <EOL> cfg_dict = conf . as_dict ( display_sensitive = True ) <EOL> self . assertEqual ( cfg_dict [ '<STR_LIT>' ] [ '<STR_LIT>' ] , '<STR_LIT>' ) <EOL> cfg_dict = conf . as_dict ( display_sensitive = True , display_source = True ) <EOL> self . assertEqual ( <EOL> cfg_dict [ '<STR_LIT>' ] [ '<STR_LIT>' ] , ( '<STR_LIT>' , '<STR_LIT>' ) ) </s>
