IPython/nbformat/tests/test_validator.py

"""
Contains tests class for validator.py
"""
#-----------------------------------------------------------------------------
#  Copyright (C) 2014  The IPython Development Team
#
#  Distributed under the terms of the BSD License.  The full license is in
#  the file COPYING, distributed as part of this software.
#-----------------------------------------------------------------------------

#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------

import os

from .base import TestsBase
from jsonschema import SchemaError
from ..current import read
from ..validator import schema_path, isvalid, validate, resolve_ref


#-----------------------------------------------------------------------------
# Classes and functions
#-----------------------------------------------------------------------------

class TestValidator(TestsBase):

    def test_schema_path(self):
        """Test that the schema path exists"""
        self.assertEqual(os.path.exists(schema_path), True)

    def test_nb2(self):
        """Test that a v2 notebook converted to v3 passes validation"""
        with self.fopen(u'test2.ipynb', u'r') as f:
            nb = read(f, u'json')
        self.assertEqual(validate(nb), [])
        self.assertEqual(isvalid(nb), True)

    def test_nb3(self):
        """Test that a v3 notebook passes validation"""
        with self.fopen(u'test3.ipynb', u'r') as f:
            nb = read(f, u'json')
        self.assertEqual(validate(nb), [])
        self.assertEqual(isvalid(nb), True)

    def test_invalid(self):
        """Test than an invalid notebook does not pass validation"""
        # this notebook has a few different errors:
        # - the name is an integer, rather than a string
        # - one cell is missing its source
        # - one cell has an invalid level
        with self.fopen(u'invalid.ipynb', u'r') as f:
            nb = read(f, u'json')
        self.assertEqual(len(validate(nb)), 3)
        self.assertEqual(isvalid(nb), False)

    def test_resolve_ref(self):
        """Test that references are correctly resolved"""
        # make sure it resolves the ref correctly
        json = {"abc": "def", "ghi": {"$ref": "/abc"}}
        resolved = resolve_ref(json)
        self.assertEqual(resolved, {"abc": "def", "ghi": "def"})

        # make sure it throws an error if the ref is not by itself
        json = {"abc": "def", "ghi": {"$ref": "/abc", "foo": "bar"}}
        with self.assertRaises(SchemaError):
            resolved = resolve_ref(json)

        # make sure it can handle json with no reference
        json = {"abc": "def"}
        resolved = resolve_ref(json)
        self.assertEqual(resolved, json)