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()