def test_create_cluster_set(self):
"""
"""
# Setup function
self.catalogue = Catalogue()
self.catalogue.data["EventID"] = np.array([1, 2, 3, 4, 5, 6])
self.catalogue.data["magnitude"] = np.array([7.0, 5.0, 5.0, 5.0, 4.0, 4.0])
selector0 = CatalogueSelector(self.catalogue)
vcl = np.array([0, 1, 1, 1, 2, 2])
cluster_set = selector0.create_cluster_set(vcl)
np.testing.assert_array_equal(cluster_set[0].data["EventID"], np.array([1]))
np.testing.assert_array_almost_equal(cluster_set[0].data["magnitude"], np.array([7.0]))
np.testing.assert_array_equal(cluster_set[1].data["EventID"], np.array([2, 3, 4]))
np.testing.assert_array_almost_equal(cluster_set[1].data["magnitude"], np.array([5.0, 5.0, 5.0]))
np.testing.assert_array_equal(cluster_set[2].data["EventID"], np.array([5, 6]))
np.testing.assert_array_almost_equal(cluster_set[2].data["magnitude"], np.array([4.0, 4.0]))
def test_create_cluster_set(self):
"""
"""
# Setup function
self.catalogue = Catalogue()
self.catalogue.data["EventID"] = np.array([1, 2, 3, 4, 5, 6])
self.catalogue.data["magnitude"] = np.array(
[7.0, 5.0, 5.0, 5.0, 4.0, 4.0])
selector0 = CatalogueSelector(self.catalogue)
vcl = np.array([0, 1, 1, 1, 2, 2])
cluster_set = selector0.create_cluster_set(vcl)
np.testing.assert_array_equal(cluster_set[0].data["EventID"],
np.array([1]))
np.testing.assert_array_almost_equal(cluster_set[0].data["magnitude"],
np.array([7.0]))
np.testing.assert_array_equal(cluster_set[1].data["EventID"],
np.array([2, 3, 4]))
np.testing.assert_array_almost_equal(cluster_set[1].data["magnitude"],
np.array([5.0, 5.0, 5.0]))
np.testing.assert_array_equal(cluster_set[2].data["EventID"],
np.array([5, 6]))
np.testing.assert_array_almost_equal(cluster_set[2].data["magnitude"],
np.array([4.0, 4.0]))
