add bitarray-based polynomial base objects

ilanschnell · Jan 20, 2021 · 48fe833 · 48fe833
1 parent 9fc76a9
commit 48fe833
Show file tree

Hide file tree

Showing 7 changed files with 776 additions and 42 deletions.
diff --git a/bitarray/__init__.py b/bitarray/__init__.py
@@ -8,9 +8,7 @@
 
 Author: Ilan Schnell
 """
-from __future__ import absolute_import
-
-from bitarray._bitarray import (bitarray, decodetree, _sysinfo,
+from bitarray._bitarray import (bitarray, decodetree, _sysinfo, tbase,
                                 get_default_endian, _set_default_endian, eval_all_terms,
                                 __version__)
 

diff --git a/bitarray/_bitarray.c b/bitarray/_bitarray.c
@@ -3359,6 +3359,11 @@ init_bitarray(void)
     Py_INCREF((PyObject *) &Bitarray_Type);
     PyModule_AddObject(m, "bitarray", (PyObject *) &Bitarray_Type);
 
+    if (PyType_Ready(&TBase_Type) < 0)
+        goto error;
+    Py_INCREF((PyObject *) &TBase_Type);
+    PyModule_AddObject(m, "tbase", (PyObject *) &TBase_Type);
+
     if (PyType_Ready(&DecodeTree_Type) < 0)
         goto error;
     Py_SET_TYPE(&DecodeTree_Type, &PyType_Type);