Fix lines missing from tests

dwhswenson · Apr 11, 2021 · d59f90e · d59f90e
1 parent eb0c208
commit d59f90e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git 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
@@ -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 = (