Formatting

Author: bdlucas1

mathics/core/atoms.py

@@ -1166,7 +1166,9 @@ def __init__(self, value, dtype=None):
     # TODO: to make it less expensive only look at first 100 bytes - ok? needed?
     def __hash__(self):
         if not self._hash:
-            self._hash = hash(("NumericArray", self.value.shape, self.value.tobytes()[:100]))
+            self._hash = hash(
+                ("NumericArray", self.value.shape, self.value.tobytes()[:100])
+            )
         return self._hash
 
     def __str__(self) -> str:
@@ -1184,6 +1186,7 @@ def default_format(self, evaluation, form) -> str:
     @property
     def items(self) -> tuple:
         from mathics.core.convert.python import from_python
+
         if len(self.value.shape) == 1:
             return tuple(from_python(item.item()) for item in self.value)
         else:
@@ -1195,15 +1198,17 @@ def element_order(self) -> tuple:
             BASIC_ATOM_NUMERICARRAY_ELT_ORDER,
             self.value.shape,
             self.value.dtype,
-            self.value.tobytes()
+            self.value.tobytes(),
         )
 
     @property
     def pattern_precedence(self) -> tuple:
         return super().pattern_precedence
 
     def sameQ(self, rhs) -> bool:
-        return isinstance(rhs, NumericArray) and numpy.array_equal(self.value, rhs.value)
+        return isinstance(rhs, NumericArray) and numpy.array_equal(
+            self.value, rhs.value
+        )
 
     def to_sympy(self, **kwargs) -> None:
         return None

test/builtin/test_numericarray.py

@@ -1,19 +1,22 @@
-from mathics.core.atoms import Integer, NumericArray, String
-from mathics.core.convert.python import from_python
 from test.helper import check_evaluation, evaluate
 
 import numpy as np
 import pytest
 
+from mathics.core.atoms import Integer, NumericArray, String
+from mathics.core.convert.python import from_python
+
 #
 # Python API tests
 #
 
+
 def test_numericarray_atom_preserves_array_reference():
     array = np.array([1, 2, 3], dtype=np.int64)
     atom = NumericArray(array)
     assert atom.value is array
 
+
 def test_numericarray_atom_preserves_equality():
     array = np.array([1, 2, 3], dtype=np.int64)
     atom = NumericArray(array, dtype=np.float64)
@@ -26,6 +29,7 @@ def test_numericarray_expression_from_python_array():
     assert isinstance(atom, NumericArray)
     assert atom.value is array
 
+
 def test_numericarray_hash():
     a = [[1, 2], [3, 4]]
     array1a = np.array(a, dtype=np.float32)
@@ -34,16 +38,20 @@ def test_numericarray_hash():
     atom1b = from_python(array1a)
     array2 = np.array(a, dtype=np.float64)
     atom2 = from_python(array2)
-    assert hash(atom1a) == hash(atom1b), "hashes of different arrays with same value should be same"
-    assert hash(atom1a) != hash(atom2), "hashes of arrays with different values should be different"
+    assert hash(atom1a) == hash(
+        atom1b
+    ), "hashes of different arrays with same value should be same"
+    assert hash(atom1a) != hash(
+        atom2
+    ), "hashes of arrays with different values should be different"
 
 
 #
 # WL tests
 # Will not work yet
 #
 
-#@pytest.mark.parametrize(
+# @pytest.mark.parametrize(
 #    ("str_expr", "str_expected"),
 #    [
 #        ("NumericArray[{{1,2},{3,4}}]", "<Integer64, 2×2>"),
@@ -56,15 +64,13 @@ def test_numericarray_hash():
 #        ("Last[NumericArray[{{1,2}, {3,4}}]]", "<Integer64, 2>"),
 #        ("Normal[NumericArray[{{1,2}, {3,4}}]]", "{{1, 2}, {3, 4}}"),
 #    ]
-#)
-#def test_basics(str_expr, str_expected):
+# )
+# def test_basics(str_expr, str_expected):
 #    check_evaluation(str_expr, str_expected, hold_expected=True)
 #
-#def test_type_conversion():
+# def test_type_conversion():
 #    expr = evaluate("NumericArray[{1,2}]")
 #    assert isinstance(expr, NumericArray)
 #    assert expr.value.dtype == np.int64
 #    expr = evaluate('NumericArray[{1,2}, "ComplexReal32"]')
 #    assert expr.value.dtype == np.complex64
-
-