add simple _print_opts tests

epistoteles · Jun 18, 2024 · 2eb72a7 · 2eb72a7
1 parent 053d44d
commit 2eb72a7
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test__print_opts.py b/tests/test__print_opts.py
@@ -0,0 +1,15 @@
+from tensorhue._print_opts import set_printoptions, PRINT_OPTS
+from tensorhue.colors import ColorScheme
+
+
+def test_default_print_opts():
+    assert PRINT_OPTS.threshold > 100
+    assert isinstance(PRINT_OPTS.colorscheme, ColorScheme)
+
+
+def test_set_printopts():
+    set_printoptions(threshold=1234)
+    assert PRINT_OPTS.threshold == 1234
+    set_printoptions(edgeitems=4)
+    assert PRINT_OPTS.edgeitems == 4
+    assert PRINT_OPTS.threshold == 1234