def test_watch_class_instance(self):
test_env = ie.InteractiveEnvironment()
ib.watch(self)
test_env.watch(self)
self.assertEqual(ie.current_env().watching(), test_env.watching())
def test_watch_a_module_by_module_object(self): test_env = ie.InteractiveEnvironment() module = importlib.import_module(_module_name) ib.watch(module) test_env.watch(module) self.assertEqual(ie.current_env().watching(), test_env.watching())
def test_watch_locals(self): # test_env serves as local var too. test_env = ie.InteractiveEnvironment() ib.watch(locals()) test_env.watch(locals()) self.assertEqual(ie.current_env().watching(), test_env.watching())
def test_watch_main_by_default(self): test_env = ie.InteractiveEnvironment() # Current Interactive Beam env fetched and the test env are 2 instances. self.assertNotEqual(id(ie.current_env()), id(test_env)) self.assertEqual(ie.current_env().watching(), test_env.watching())
def test_watch_a_module_by_name(self): test_env = ie.InteractiveEnvironment() ib.watch(_module_name) test_env.watch(_module_name) self.assertEqual(ie.current_env().watching(), test_env.watching())
def test_get_gcs_cache_dir_invalid_path(self):
env = ie.InteractiveEnvironment()
p = beam.Pipeline()
cache_root = 'gs://'
with self.assertRaises(ValueError):
env._get_gcs_cache_dir(p, cache_root)
def test_get_sql_chain_setting_user_pipeline(self):
env = ie.InteractiveEnvironment()
p = beam.Pipeline()
chain = env.get_sql_chain(p, set_user_pipeline=True)
self.assertIs(chain.user_pipeline, p)
def test_get_empty_sql_chain(self):
env = ie.InteractiveEnvironment()
p = beam.Pipeline()
chain = env.get_sql_chain(p)
self.assertIsNotNone(chain)
self.assertEqual(chain.nodes, {})
def test_get_cache_manager_creates_cache_manager_if_absent(self):
env = ie.InteractiveEnvironment()
dummy_pipeline = beam.Pipeline()
self.assertIsNone(env.get_cache_manager(dummy_pipeline))
self.assertIsNotNone(
env.get_cache_manager(dummy_pipeline, create_if_absent=True))
def test_cleanup_registered_when_creating_new_env(self):
with patch('atexit.register') as mocked_atexit:
_ = ie.InteractiveEnvironment()
mocked_atexit.assert_called_once()
