1 files changed, 34 insertions, 0 deletions
diff --git a/_test/test_int.py b/_test/test_int.py
new file mode 100644
index 0000000..d409746
--- /dev/null
+++ b/_test/test_int.py
@@ -0,0 +1,34 @@
+# coding: utf-8
+
+import pytest  # NOQA
+
+from .roundtrip import dedent, round_trip_dump, round_trip_load
+
+# http://yaml.org/type/int.html is where underscores in integers are defined
+
+
+class TestBinHexOct:
+    def test_calculate(self):
+        # make sure type, leading zero(s) and underscore are preserved
+        s = dedent(
+            """\
+        - 42
+        - 0b101010
+        - 0x_2a
+        - 0x2A
+        - 0o00_52
+        """
+        )
+        d = round_trip_load(s)
+        for idx, elem in enumerate(d):
+            elem -= 21
+            d[idx] = elem
+        for idx, elem in enumerate(d):
+            elem *= 2
+            d[idx] = elem
+        for idx, elem in enumerate(d):
+            t = elem
+            elem **= 2
+            elem //= t
+            d[idx] = elem
+        assert round_trip_dump(d) == s