Initial commit

tests/test_tokfmt.py

@@ -0,0 +1,52 @@
+import pytest
+
+from cxxheaderparser.lexer import Lexer
+from cxxheaderparser.tokfmt import tokfmt
+
+
+@pytest.mark.parametrize(
+    "instr",
+    [
+        "int",
+        "unsigned int",
+        "::uint8_t",
+        "void *",
+        "const char *",
+        "const char[]",
+        "void * (*)()",
+        "void (*)(void * buf, int buflen)",
+        "void (* fnType)(void * buf, int buflen)",
+        "TypeName<int, void>& x",
+        "vector<string>&",
+        "std::vector<Pointer *> *",
+        "Alpha::Omega",
+        "Convoluted::Nested::Mixin",
+        "std::function<void ()>",
+        "std::shared_ptr<std::function<void (void)>>",
+        "tr1::shared_ptr<SnailTemplateClass<SnailNamespace::SnailClass>>",
+        "std::map<unsigned, std::pair<unsigned, SnailTemplateClass<SnailNamespace::SnailClass>>>",
+        "std::is_base_of<std::random_access_iterator_tag, IteratorCategoryT>::value",
+        "const char&&",
+        "something<T, U>{1, 2, 3}",
+        "operator-=",
+        "operator[]",
+        "operator*",
+        "operator>=",
+    ],
+)
+def test_tokfmt(instr):
+    """
+    Each input string is exactly what the output of tokfmt should be
+    """
+    toks = []
+    lexer = Lexer("")
+    lexer.input(instr)
+
+    while True:
+        tok = lexer.token_eof_ok()
+        if not tok:
+            break
+
+        toks.append(tok)
+
+    assert tokfmt(toks) == instr