diff --git a/tests/parse/test_note.py b/tests/parse/test_note.py
index 03fb66e..12e5e92 100644
--- a/tests/parse/test_note.py
+++ b/tests/parse/test_note.py
@@ -1,6 +1,20 @@
 # SPDX-License-Identifier: LGPL-2.1-only
 # Copyright 2022 Jookia <contact@jookia.org>
 
+# Note syntax consists of the following:
+# - StartNote
+# - Optionally any token that isn't EndNote
+# - EndNote
+#
+# Parsing gives the following:
+# No value
+#
+# The following error contexts are used:
+# PARSE_NOTE - Used when parsing the file
+#
+# The following parse errors are generated:
+# FOUND_ENDNOTE - When a stray EndNote token is found
+
 from hypothesis import assume, given
 from hypothesis.strategies import (
     composite,
@@ -25,6 +39,10 @@
     draw_token_random,
 )
 
+#
+# Helper functions
+#
+
 
 # Static tokens representing a note
 def static_note_tokens():
@@ -69,6 +87,11 @@
     return all_tokens
 
 
+#
+# Test functions
+#
+
+
 # Tests skip_note works correctly
 # We expect the following behaviour:
 # - No value is returned