mirror of
https://github.com/GoogleCloudPlatform/vertex-ai-samples.git
synced 2026-09-26 14:42:04 +00:00
Andrew FerlitschandGitHub
b002e7ba2c
* tune: selecting notebooks * test: add test for select_notebook * fix: random algo * test: convert to pytest * fix: review * fix: remove path .. * fix: assert
34 lines
764 B
Python
34 lines
764 B
Python
import sys
|
|
|
|
from execute_changed_notebooks_helper import (load_results, select_notebook)
|
|
|
|
|
|
def test_load_results():
|
|
bucket: str = "cloud-build-notebooks-presubmit"
|
|
bucket_file: str = "build_results"
|
|
|
|
accum = load_results(bucket, bucket_file)
|
|
|
|
print(accum)
|
|
|
|
assert len(accum) > 0
|
|
|
|
def test_select_notebook():
|
|
bucket: str = "cloud-build-notebooks-presubmit"
|
|
bucket_file: str = "build_results"
|
|
|
|
accum = load_results(bucket, bucket_file)
|
|
|
|
n_select = 0
|
|
n_notselect = 0
|
|
for notebook in accum:
|
|
if select_notebook(notebook, accum, 50):
|
|
n_select += 1
|
|
else:
|
|
n_notselect += 1
|
|
|
|
print(f"SELECTED {n_select}, NOT SELECTED {n_notselect}")
|
|
|
|
assert n_select > 0
|
|
assert n_notselect > 0
|