def test_has_cluster_unknown_with_unknown_type(self):
self.assertTrue(
WordClusters('brown', 'brown_file',
unknown_type="<UNK>").hasCluster('unknown type'))
def test_has_cluster_unknown_without_unknown_type(self):
self.assertFalse(
WordClusters('brown', 'brown_file').hasCluster('unknown type'))
def test_is_oov(self):
self.assertTrue(
WordClusters('brown', 'brown_file',
unknown_type="<UNK>").isOOV('unknown type'))
def test_has_cluster_known(self):
self.assertTrue(
WordClusters('brown', 'brown_file',
unknown_type="<UNK>").hasCluster('cat'))
def test_clusters_unknown_not_in_same_cluster(self):
self.assertFalse(
WordClusters('brown', 'brown_file',
unknown_type="<UNK>").inSameCluster(
'mouse', 'unknown type'))
def test_clusters_unknown_without_unknown_type(self):
self.assertTrue(
WordClusters('brown',
'brown_file').inSameCluster('unknown type',
'another unknown'))
def test_clusters_unknown_in_same_cluster(self):
self.assertTrue(
WordClusters('brown', 'brown_file',
unknown_type="<UNK>").inSameCluster(
'cat', 'unknown type'))
def test_clusters_unknown_not_in_vocabulary(self):
with self.assertRaises(ValueError):
WordClusters('brown', 'brown_file', unknown_type="unknown type")
def test_clusters_not_in_same_cluster(self):
self.assertFalse(
WordClusters('brown', 'brown_file').inSameCluster('cat', 'mouse'))
def test_clusters_in_same_cluster(self):
self.assertTrue(
WordClusters('brown', 'brown_file').inSameCluster('cat', 'dog'))
def test_cluster_type_not_supported(self):
with self.assertRaises(ValueError):
WordClusters('Unknown type', 'brown_file')