def test_getattrd(self):
"""Testing getattrd."""
with self.assertRaises(AttributeError): at.getattrd(int, 'a')
assert at.getattrd(int, 'a', default=None) is None
assert at.getattrd(int, '__class__.__name__') == "type"
assert at.hasattrd(int, '__class__.__name__')
assert not at.hasattrd(int, 'foobar.__name__')
def _sortby_labelfile_list(self, labelfile_list, func_or_string, reverse=False, unpack=False):
"""
Return: list of (label, abifile, param) tuples where param is obtained via ``func_or_string``.
or labels, abifiles, params if ``unpack``
"""
if not func_or_string:
# Catch None or empty
items = [(label, abifile, label) for (label, abifile) in labelfile_list]
if not unpack:
return items
else:
return [t[0] for t in items], [t[1] for t in items], [t[2] for t in items]
elif callable(func_or_string):
items = [(label, abifile, func_or_string(abifile)) for (label, abifile) in labelfile_list]
else:
# Assume string and attribute with the same name.
# try in abifile.params if not hasattrd(abifile, func_or_string)
self.is_sortable(func_or_string, raise_exc=True)
if hasattrd(self.abifiles[0], func_or_string):
items = [(label, abifile, getattrd(abifile, func_or_string)) for (label, abifile) in labelfile_list]
else:
items = [(label, abifile, abifile.params[func_or_string]) for (label, abifile) in labelfile_list]
items = sorted(items, key=lambda t: t[2], reverse=reverse)
if not unpack:
return items
else:
return [t[0] for t in items], [t[1] for t in items], [t[2] for t in items]
def _sortby_labelfile_list(self, labelfile_list, func_or_string, reverse=False, unpack=False):
"""
Return: list of (label, abifile, param) tuples where param is obtained via ``func_or_string``.
or labels, abifiles, params if ``unpack``
"""
if not func_or_string:
# Catch None or empty
items = [(label, abifile, label) for (label, abifile) in labelfile_list]
if not unpack:
return items
else:
return [t[0] for t in items], [t[1] for t in items], [t[2] for t in items]
elif callable(func_or_string):
items = [(label, abifile, func_or_string(abifile)) for (label, abifile) in labelfile_list]
else:
# Assume string and attribute with the same name.
# try in abifile.params if not hasattrd(abifile, func_or_string)
self.is_sortable(func_or_string, raise_exc=True)
if hasattrd(self.abifiles[0], func_or_string):
items = [(label, abifile, getattrd(abifile, func_or_string)) for (label, abifile) in labelfile_list]
else:
items = [(label, abifile, abifile.params[func_or_string]) for (label, abifile) in labelfile_list]
items = sorted(items, key=lambda t: t[2], reverse=reverse)
if not unpack:
return items
else:
return [t[0] for t in items], [t[1] for t in items], [t[2] for t in items]
def group_and_sortby(self, hue, func_or_string):
"""
Group files by ``hue`` and, inside each group` sort items by ``func_or_string``.
Args:
hue: Variable that define subsets of the data, which will be drawn on separate lines.
Accepts callable or string
If string, it's assumed that the abifile has an attribute with the same name and getattr is invoked.
Dot notation is also supported e.g. hue="structure.formula" --> abifile.structure.formula
If callable, the output of hue(abifile) is used.
func_or_string: Either None, string, callable defining the quantity to be used for sorting.
If string, it's assumed that the abifile has an attribute with the same name and getattr is invoked.
If callable, the output of func_or_string(abifile) is used.
If None, no sorting is performed.
Return: List of :class:`HueGroup` instance.
"""
# Group by hue.
# This is the section in which we support: callable, abifile.attr.name syntax or abifile.params["key"]
items = list(self.items())
if callable(hue):
key = lambda t: hue(t[1])
else:
# Assume string.
if hasattrd(self.abifiles[0], hue):
key = lambda t: getattrd(t[1], hue)
else:
# Try in abifile.params
if hasattr(self.abifiles[0],
"params") and hue in self.abifiles[0].params:
key = lambda t: t[1].params[hue]
else:
raise TypeError("""\
Cannot interpret hue argument of type `%s` and value `%s`.
Expecting callable or attribute name or key in abifile.params""" %
(type(hue), str(hue)))
groups = []
for hvalue, labelfile_list in sort_and_groupby(items, key=key):
# Use func_or_string to sort each group
labels, abifiles, xvalues = self._sortby_labelfile_list(
labelfile_list, func_or_string, unpack=True)
groups.append(HueGroup(hvalue, xvalues, abifiles, labels))
return groups
def group_and_sortby(self, hue, func_or_string):
"""
Group files by ``hue`` and, inside each group` sort items by ``func_or_string``.
Args:
hue: Variable that define subsets of the data, which will be drawn on separate lines.
Accepts callable or string
If string, it's assumed that the abifile has an attribute with the same name and getattr is invoked.
Dot notation is also supported e.g. hue="structure.formula" --> abifile.structure.formula
If callable, the output of hue(abifile) is used.
func_or_string: Either None, string, callable defining the quantity to be used for sorting.
If string, it's assumed that the abifile has an attribute with the same name and getattr is invoked.
If callable, the output of func_or_string(abifile) is used.
If None, no sorting is performed.
Return: List of :class:`HueGroup` instance.
"""
# Group by hue.
# This is the section in which we support: callable, abifile.attr.name syntax or abifile.params["key"]
items = list(self.items())
if callable(hue):
key = lambda t: hue(t[1])
else:
# Assume string.
if hasattrd(self.abifiles[0], hue):
key = lambda t: getattrd(t[1], hue)
else:
# Try in abifile.params
if hasattr(self.abifiles[0], "params") and hue in self.abifiles[0].params:
key = lambda t: t[1].params[hue]
else:
raise TypeError("""\
Cannot interpret hue argument of type `%s` and value `%s`.
Expecting callable or attribute name or key in abifile.params""" % (type(hue), str(hue)))
groups = []
for hvalue, labelfile_list in sort_and_groupby(items, key=key):
# Use func_or_string to sort each group
labels, abifiles, xvalues = self._sortby_labelfile_list(labelfile_list, func_or_string, unpack=True)
groups.append(HueGroup(hvalue, xvalues, abifiles, labels))
return groups