def test_unique_is_not_present(self):
"""Test that Nones are returned if no unique site ID coordinate is
present on the neighbour cube."""
self.neighbour_cube.remove_coord("met_office_site_id")
plugin = SpotExtraction()
result = plugin.check_for_unique_id(self.neighbour_cube)
self.assertIsNone(result)
def test_unique_is_present(self):
"""Test that the IDs and coordinate name are returned if a unique site
ID coordinate is present on the neighbour cube."""
plugin = SpotExtraction()
result = plugin.check_for_unique_id(self.neighbour_cube)
self.assertArrayEqual(result[0], self.unique_site_id)
self.assertEqual(result[1], self.unique_site_id_key)
