From d59f90e892cd9bc6122238f9072919df7cfd5df4 Mon Sep 17 00:00:00 2001 From: "David W.H. Swenson" Date: Sun, 11 Apr 2021 17:44:03 +0200 Subject: [PATCH] Fix lines missing from tests --- contact_map/contact_count.py | 3 ++- contact_map/contact_map.py | 8 ++------ 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/contact_map/contact_count.py b/contact_map/contact_count.py index 9e2f1be..9b8577f 100644 --- a/contact_map/contact_count.py +++ b/contact_map/contact_count.py @@ -204,7 +204,8 @@ def plot(self, cmap='seismic', vmin=-1.0, vmax=1.0, with_colorbar=True, # Check the number of pixels of the figure self._check_number_of_pixels(fig) - self.plot_axes(ax=ax, cmap=cmap, vmin=vmin, vmax=vmax) + self.plot_axes(ax=ax, cmap=cmap, vmin=vmin, vmax=vmax, + with_colorbar=with_colorbar) return (fig, ax) diff --git a/contact_map/contact_map.py b/contact_map/contact_map.py index a8599f8..ffcdbae 100644 --- a/contact_map/contact_map.py +++ b/contact_map/contact_map.py @@ -600,15 +600,11 @@ def _contact_map(self, trajectory, frame_number, residue_query_atom_idxs, @property def atom_contacts(self): - n_atoms = self.topology.n_atoms - return ContactCount(self._atom_contacts, self.topology.atom, - n_atoms, n_atoms) + raise NotImplementedError() @property def residue_contacts(self): - n_res = self.topology.n_residues - return ContactCount(self._residue_contacts, self.topology.residue, - n_res, n_res) + raise NotImplementedError() CONTACT_MAP_ERROR = (