@@ -460,108 +460,4 @@ def test__grid_2d_native_from():
460460 [[- 1.0 , - 1.0 ], [- 2.0 , - 2.0 ], [0.0 , 0.0 ], [- 3.0 , - 3.0 ]],
461461 ]
462462 )
463- ).all ()
464-
465- grid_slim = np .array (
466- [
467- [1.0 , 1.0 ],
468- [1.0 , 1.0 ],
469- [1.0 , 1.0 ],
470- [1.0 , 1.0 ],
471- [2.0 , 2.0 ],
472- [2.0 , 2.0 ],
473- [2.0 , 2.0 ],
474- [2.0 , 2.0 ],
475- [3.0 , 3.0 ],
476- [3.0 , 3.0 ],
477- [3.0 , 3.0 ],
478- [4.0 , 4.0 ],
479- ]
480- )
481-
482-
483- def test__grid_2d_slim_upscaled_from ():
484- grid_slim = np .array ([[1.0 , 1.0 ]])
485-
486- grid_upscaled_2d = aa .util .grid_2d .grid_2d_slim_upscaled_from (
487- grid_slim = grid_slim , upscale_factor = 1 , pixel_scales = (2.0 , 2.0 )
488- )
489-
490- assert (grid_upscaled_2d == np .array ([[1.0 , 1.0 ]])).all ()
491-
492- grid_upscaled_2d = aa .util .grid_2d .grid_2d_slim_upscaled_from (
493- grid_slim = grid_slim , upscale_factor = 2 , pixel_scales = (2.0 , 2.0 )
494- )
495-
496- assert (
497- grid_upscaled_2d == np .array ([[1.5 , 0.5 ], [1.5 , 1.5 ], [0.5 , 0.5 ], [0.5 , 1.5 ]])
498- ).all ()
499-
500- grid_slim = np .array ([[1.0 , 1.0 ], [1.0 , 3.0 ]])
501-
502- grid_upscaled_2d = aa .util .grid_2d .grid_2d_slim_upscaled_from (
503- grid_slim = grid_slim , upscale_factor = 2 , pixel_scales = (2.0 , 2.0 )
504- )
505-
506- assert (
507- grid_upscaled_2d
508- == np .array (
509- [
510- [1.5 , 0.5 ],
511- [1.5 , 1.5 ],
512- [0.5 , 0.5 ],
513- [0.5 , 1.5 ],
514- [1.5 , 2.5 ],
515- [1.5 , 3.5 ],
516- [0.5 , 2.5 ],
517- [0.5 , 3.5 ],
518- ]
519- )
520- ).all ()
521-
522- grid_slim = np .array ([[1.0 , 1.0 ], [3.0 , 1.0 ]])
523-
524- grid_upscaled_2d = aa .util .grid_2d .grid_2d_slim_upscaled_from (
525- grid_slim = grid_slim , upscale_factor = 2 , pixel_scales = (2.0 , 2.0 )
526- )
527-
528- assert (
529- grid_upscaled_2d
530- == np .array (
531- [
532- [1.5 , 0.5 ],
533- [1.5 , 1.5 ],
534- [0.5 , 0.5 ],
535- [0.5 , 1.5 ],
536- [3.5 , 0.5 ],
537- [3.5 , 1.5 ],
538- [2.5 , 0.5 ],
539- [2.5 , 1.5 ],
540- ]
541- )
542- ).all ()
543-
544- grid_slim = np .array ([[1.0 , 1.0 ]])
545-
546- grid_upscaled_2d = aa .util .grid_2d .grid_2d_slim_upscaled_from (
547- grid_slim = grid_slim , upscale_factor = 2 , pixel_scales = (3.0 , 2.0 )
548- )
549-
550- assert (
551- grid_upscaled_2d
552- == np .array ([[1.75 , 0.5 ], [1.75 , 1.5 ], [0.25 , 0.5 ], [0.25 , 1.5 ]])
553- ).all ()
554-
555- grid_upscaled_2d = aa .util .grid_2d .grid_2d_slim_upscaled_from (
556- grid_slim = grid_slim , upscale_factor = 3 , pixel_scales = (2.0 , 2.0 )
557- )
558-
559- assert grid_upscaled_2d [0 ] == pytest .approx (np .array ([1.666 , 0.333 ]), 1.0e-2 )
560- assert grid_upscaled_2d [1 ] == pytest .approx (np .array ([1.666 , 1.0 ]), 1.0e-2 )
561- assert grid_upscaled_2d [2 ] == pytest .approx (np .array ([1.666 , 1.666 ]), 1.0e-2 )
562- assert grid_upscaled_2d [3 ] == pytest .approx (np .array ([1.0 , 0.333 ]), 1.0e-2 )
563- assert grid_upscaled_2d [4 ] == pytest .approx (np .array ([1.0 , 1.0 ]), 1.0e-2 )
564- assert grid_upscaled_2d [5 ] == pytest .approx (np .array ([1.0 , 1.666 ]), 1.0e-2 )
565- assert grid_upscaled_2d [6 ] == pytest .approx (np .array ([0.333 , 0.333 ]), 1.0e-2 )
566- assert grid_upscaled_2d [7 ] == pytest .approx (np .array ([0.333 , 1.0 ]), 1.0e-2 )
567- assert grid_upscaled_2d [8 ] == pytest .approx (np .array ([0.333 , 1.666 ]), 1.0e-2 )
463+ ).all ()
0 commit comments