Flatten the package a bit for end users.

Author: Julian

README.rst

@@ -30,7 +30,7 @@ Here's a simple example:
 
 .. code-block:: python
 
-  from jsonschema_lexer.lexer import JSONSchemaLexer
+  from jsonschema_lexer import JSONSchemaLexer
 
   from rich.console import Console
   from rich.syntax import Syntax

jsonschema_lexer/__init__.py

@@ -1,3 +1,7 @@
 """
-Provides the JSONSchema Lexer.
+A JSON Schema lexer for Pygments.
 """
+
+from jsonschema_lexer._lexer import JSONSchemaLexer
+
+__all__ = ["JSONSchemaLexer"]

jsonschema_lexer/lexer.py renamed to jsonschema_lexer/_lexer.py

@@ -1,13 +1,11 @@
 """
-    JSON Schema Lexer tests
+JSON Schema Lexer tests
 """
 
 from pygments.token import Token
 import pytest
 
-from jsonschema_lexer.lexer import JSONSchemaLexer
-
-# Test helpers.
+from jsonschema_lexer import JSONSchemaLexer
 
 
 @pytest.fixture()
@@ -26,27 +24,24 @@ def keywords():
 
 
 def assert_single_token(lexer, s, token):
-    """Show that a given string generates only one token."""
+    """
+    Assert a given string generates only one token.
+    """
     tokens = list(lexer.get_tokens_unprocessed(s))
     assert len(tokens) == 1
     assert s == tokens[0][2]
     assert token == tokens[0][1]
 
 
 def assert_tokens(lexer, string, expected_tokens):
-    """Show that a given string generates the expected tokens."""
+    """
+    Assert a given string generates the expected tokens.
+    """
     tokens = list(lexer.get_tokens_unprocessed(string))
     parsed_tokens = [t[1] for t in tokens]
     assert parsed_tokens == expected_tokens
 
 
-# Tests
-
-
-def test_it_imports():
-    import jsonschema_lexer  # noqa: F401
-
-
 def test_data_type_tokens(lexer, data_types):
     for data_type in data_types:
         assert_single_token(lexer, data_type, Token.Name.Decorator)