from pylabelbuddy import _annotations_notebook


def test_annotations_notebook(root, annotations_mock, dataset_mock):
    nb = _annotations_notebook.AnnotationsNotebook(
        root, annotations_mock, dataset_mock
    )
    nb.change_database()
    assert nb.notebook.index(nb.notebook.select()) == 2
    nb.go_to_annotations()
    assert nb.notebook.index(nb.notebook.select()) == 0

[((5, 9, 7, 5), 'pylabelbuddy._annotations_notebook.AnnotationsNotebook', '_annotations_notebook.AnnotationsNotebook', ({(6, 8, 6, 12): 'root', (6, 14, 6, 30): 'annotations_mock', (6, 32, 6, 44): 'dataset_mock'}, {}), '(root, annotations_mock, dataset_mock)', False, 'from pylabelbuddy import _annotations_notebook\n')]