def test_extract_extrametadata_disable_bad_regex(): try: metadata = extract_extrametadata(nbformat.read(LINT_DISABLE_NB, 4), noqa_regex=r"^# noqa notebook:\s*.*$") except ValueError as e: assert e.args[ 0] == "noqa_regex must contain one capture group (specifying the rule)" else: assert False, "should have raised a ValueError"
def _metadata(nb, what): extra_metadata = extract_extrametadata(nbformat.read(nb, 4)) return extra_metadata[what]
def test_extract_extrametadata_disable_cells_count(): metadata = extract_extrametadata(nbformat.read(LINT_DISABLE_NB, 4), noqa_regex=r"^# noqa notebook:\s*(.*)$") assert metadata['noqa'] == {'cells_per_notebook'}
def test_extract_extrametadata_disable_notpresent(): metadata = extract_extrametadata( nbformat.read(LINT_DISABLE_NB, 4), noqa_regex=r"^# don't noqa notebook:\s*(.*)$") assert len(metadata['noqa']) == 0
def test_extract_extrametadata_disable_none(): metadata = extract_extrametadata(nbformat.read(LINT_DISABLE_NB, 4)) assert len(metadata['noqa']) == 0
def test_get_coverage_nb(): assert get_coverage(extract_extrametadata(nbformat.read(COVERAGE_NB, 4))) == 25
def test_extract_extrametadata_disable_cells_count_in_empty_cell(): metadata = extract_extrametadata( nbformat.read(LINT_DISABLE_NB_EMPTY_CELL, 4), noqa_regex=r"^# noqa notebook:\s*(.*)$", ) assert metadata["noqa"] == {"cells_per_notebook"}