def test_get_n_amps_2d(): n_amps = 4 test_image = FakeImage(n_amps=n_amps) assert test_image.get_n_amps() == n_amps
def test_get_n_amps_3d(): test_image = FakeImage() assert test_image.get_n_amps() == 1