class test_ImageObject():
def setUp(self):
self.imageobject = ImageObject({}, "")
def _load_file(self, fname):
with open(fname, "rb") as f:
dm = DigitalMicrographReader(f)
dm.parse_file()
self.imdict = dm.get_image_dictionaries()
return [ImageObject(imdict, fname) for imdict in self.imdict]
def test_get_microscope_name(self):
fname = os.path.join(my_path, "dm3_2D_data",
"test_diffraction_pattern_tags_removed.dm3")
images = self._load_file(fname)
image = images[0]
# Should return None because the tags are missing
assert image._get_microscope_name(image.imdict.ImageTags) is None
fname = os.path.join(my_path, "dm3_2D_data",
"test_diffraction_pattern.dm3")
images = self._load_file(fname)
image = images[0]
assert image._get_microscope_name(
image.imdict.ImageTags) == "FEI Tecnai"
def test_get_date(self):
assert self.imageobject._get_date("11/13/2016") == "2016-11-13"
def test_get_time(self):
assert self.imageobject._get_time("6:56:37 pm") == "18:56:37"
def test_parse_string(self):
assert self.imageobject._parse_string("") is None
assert self.imageobject._parse_string("string") is "string"
class TestImageObject():
def setup_method(self, method):
self.imageobject = ImageObject({}, "")
def _load_file(self, fname):
with open(fname, "rb") as f:
dm = DigitalMicrographReader(f)
dm.parse_file()
self.imdict = dm.get_image_dictionaries()
return [ImageObject(imdict, fname) for imdict in self.imdict]
def test_get_microscope_name(self):
fname = os.path.join(MY_PATH, "dm3_2D_data",
"test_diffraction_pattern_tags_removed.dm3")
images = self._load_file(fname)
image = images[0]
# Should return None because the tags are missing
assert image._get_microscope_name(image.imdict.ImageTags) is None
fname = os.path.join(MY_PATH, "dm3_2D_data",
"test_diffraction_pattern.dm3")
images = self._load_file(fname)
image = images[0]
assert image._get_microscope_name(
image.imdict.ImageTags) == "FEI Tecnai"
def test_get_date(self):
assert self.imageobject._get_date("11/13/2016") == "2016-11-13"
def test_get_time(self):
assert self.imageobject._get_time("6:56:37 pm") == "18:56:37"
def test_parse_string(self):
assert self.imageobject._parse_string("") is None
assert self.imageobject._parse_string("string") is "string"
def _load_file(self, fname):
with open(fname, "rb") as f:
dm = DigitalMicrographReader(f)
dm.parse_file()
self.imdict = dm.get_image_dictionaries()
return [ImageObject(imdict, fname) for imdict in self.imdict]
def setUp(self):
self.imageobject = ImageObject({}, "")
def setup_method(self, method):
self.imageobject = ImageObject({}, "")
def setup_method(self, method):
self.imageobject = ImageObject({}, "")
