diffpy.morph/tests/test_morphscale.py at main · diffpy/diffpy.morph · GitHub

Repository navigation

#!/usr/bin/env python

import os

import numpy

import pytest

from diffpy.morph.morphs.morphscale import MorphScale

# useful variables

thisfile = locals().get("__file__", "file.py")

tests_dir = os.path.dirname(os.path.abspath(thisfile))

# testdata_dir = os.path.join(tests_dir, 'testdata')

class TestMorphScale:

@pytest.fixture

def setup(self):

self.x_morph = numpy.arange(0.01, 5, 0.01)

self.y_morph = numpy.ones_like(self.x_morph)

self.x_target = numpy.arange(0.01, 5, 0.01)

self.y_target = 3 * numpy.ones_like(self.x_target)

return

def test_morph(self, setup):

"""Check MorphScale.morph()"""

config = {"scale": 2.0}

morph = MorphScale(config)

x_morph, y_morph, x_target, y_target = morph(

self.x_morph, self.y_morph, self.x_target, self.y_target

)

assert numpy.allclose(2 * self.y_morph, y_morph)

assert numpy.allclose(self.y_target, y_target)

return