<s> import unittest <EOL> from googkit . compat . unittest import mock <EOL> from googkit . lib . environment import Environment <EOL> class TestEnvironmtnt ( unittest . TestCase ) : <EOL> def test_init ( self ) : <EOL> cwd = '<STR_LIT>' <EOL> mock_argument = mock . MagicMock ( ) <EOL> mock_tree = mock . MagicMock ( ) <EOL> env = Environment ( cwd , mock_argument , mock_tree ) <EOL> self . assertEqual ( env . cwd , cwd ) <EOL> self . assertEqual ( env . argument , mock_argument ) <EOL> self . assertEqual ( env . tree , mock_tree ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> unittest . main ( ) </s>
