| 777 | self.assertEqual(os.environ['var'], 'val2') |
| 778 | |
| 779 | def test_env_get_set_complex(self): |
| 780 | env = _ip.magic("env var 'val1 '' 'val2") |
| 781 | self.assertEqual(env, None) |
| 782 | self.assertEqual(os.environ['var'], "'val1 '' 'val2") |
| 783 | self.assertEqual(_ip.magic("env var"), "'val1 '' 'val2") |
| 784 | env = _ip.magic('env var=val2 val3="val4') |
| 785 | self.assertEqual(env, None) |
| 786 | self.assertEqual(os.environ['var'], 'val2 val3="val4') |
| 787 | |
| 788 | def test_env_set_bad_input(self): |
| 789 | self.assertRaises(UsageError, lambda: _ip.magic("set_env var")) |