def test_num_of_workers_3(): eq_(bdl_utils.get_num_of_workers(bdl_utils.convert("""#!benchDL pool(size = 17K, worker_type = dummy_worker): do_stuff(1,2) pool(size = 13M, worker_type = dummy_worker): do_stuff(1,2)""", {})), 13017000)
def test_num_of_workers_2(): eq_(bdl_utils.get_num_of_workers(bdl_utils.convert("""#!benchDL pool(size = var("num2", 3), worker_type = dummy_worker): do_stuff(1,2) pool(size = var("num", 3), worker_type = dummy_worker): do_stuff(1,2)""", {"num": 7})), 10)