Merge pull request #1331 from jhlegarreta/FixNpCompatWarning

effigies · web-flow · commit 0ebf3901e960 · 2024-06-27T03:05:39.000-04:00
RF: Remove unnecessary call to `asbytes` for `b`-prepended strings
diff --git a/nibabel/streamlines/tests/test_streamlines.py b/nibabel/streamlines/tests/test_streamlines.py
@@ -6,7 +6,6 @@
 
 import numpy as np
 import pytest
-from numpy.compat.py3k import asbytes
 
 import nibabel as nib
 from nibabel.testing import clear_and_catch_warnings, data_path, error_warnings
@@ -95,7 +94,7 @@ def test_is_supported_detect_format(tmp_path):
     # Valid file without extension
     for tfile_cls in FORMATS.values():
         f = BytesIO()
-        f.write(asbytes(tfile_cls.MAGIC_NUMBER))
+        f.write(tfile_cls.MAGIC_NUMBER)
         f.seek(0, os.SEEK_SET)
         assert nib.streamlines.is_supported(f)
         assert nib.streamlines.detect_format(f) is tfile_cls
@@ -104,7 +103,7 @@ def test_is_supported_detect_format(tmp_path):
     for tfile_cls in FORMATS.values():
         fpath = tmp_path / 'test.txt'
         with open(fpath, 'w+b') as f:
-            f.write(asbytes(tfile_cls.MAGIC_NUMBER))
+            f.write(tfile_cls.MAGIC_NUMBER)
             f.seek(0, os.SEEK_SET)
             assert nib.streamlines.is_supported(f)
             assert nib.streamlines.detect_format(f) is tfile_cls
