44from asdf .testing import helpers
55from astropy import units
66from astropy .units import Quantity
7- from astropy .utils .introspection import minversion
87from numpy .testing import assert_array_equal
98
109from asdf_astropy .tests .versions import ASTROPY_GE_7_1
1110
1211
13- def asdf_open_memory_mapping_kwarg (memmap : bool ) -> dict :
14- if minversion ("asdf" , "3.1.0" ):
15- return {"memmap" : memmap }
16- return {"copy_arrays" : not memmap }
17-
18-
1912def create_quantities ():
2013 return [
2114 # Scalar:
@@ -104,7 +97,7 @@ def test_memmap(tmp_path):
10497 af .write_to (file_path )
10598
10699 # Update a value in the ASDF file
107- with asdf .open (file_path , mode = "rw" , ** asdf_open_memory_mapping_kwarg ( memmap = True ) ) as af :
100+ with asdf .open (file_path , mode = "rw" , memmap = True ) as af :
108101 assert (af .tree ["quantity" ] == quantity ).all ()
109102 assert af .tree ["quantity" ][- 1 , - 1 ] != new_value
110103
@@ -114,7 +107,7 @@ def test_memmap(tmp_path):
114107 assert (af .tree ["quantity" ] != quantity ).any ()
115108 assert (af .tree ["quantity" ] == new_quantity ).all ()
116109
117- with asdf .open (file_path , mode = "rw" , ** asdf_open_memory_mapping_kwarg ( memmap = True ) ) as af :
110+ with asdf .open (file_path , mode = "rw" , memmap = True ) as af :
118111 assert af .tree ["quantity" ][- 1 , - 1 ] == new_value
119112 assert (af .tree ["quantity" ] != quantity ).any ()
120113 assert (af .tree ["quantity" ] == new_quantity ).all ()
@@ -137,7 +130,7 @@ def test_no_memmap(tmp_path):
137130 af .write_to (file_path )
138131
139132 # Update a value in the ASDF file
140- with asdf .open (file_path , mode = "rw" , ** asdf_open_memory_mapping_kwarg ( memmap = False ) ) as af :
133+ with asdf .open (file_path , mode = "rw" , memmap = False ) as af :
141134 assert (af .tree ["quantity" ] == quantity ).all ()
142135 assert af .tree ["quantity" ][- 1 , - 1 ] != new_value
143136
@@ -147,7 +140,7 @@ def test_no_memmap(tmp_path):
147140 assert (af .tree ["quantity" ] != quantity ).any ()
148141 assert (af .tree ["quantity" ] == new_quantity ).all ()
149142
150- with asdf .open (file_path , mode = "rw" , ** asdf_open_memory_mapping_kwarg ( memmap = False ) ) as af :
143+ with asdf .open (file_path , mode = "rw" , memmap = False ) as af :
151144 assert af .tree ["quantity" ][- 1 , - 1 ] != new_value
152145 assert (af .tree ["quantity" ] != new_quantity ).any ()
153146 assert (af .tree ["quantity" ] == quantity ).all ()
0 commit comments