def test_get_properties(self):
all_properties = programfiles.Properties()
self.assertEqual(all_properties.properties, [])
a_property = programfiles.Property("What a nice property",
"Property 1")
all_properties.add_property(a_property)
self.assertEqual(all_properties.properties[0], a_property)
def test_vectorize(self):
a_property = programfiles.Property(self._txt, "Property 1")
data = programfiles.PropertyData()
data.extract(self._pos_tags,
self._noun_categories,
a_property,
add_extra_nouns=True)
a_property.set_data(data)
a_property.vectorize(penalty=0.5)
vectorized_target = Counter({
('apartment', None): 0.5,
('apartment', 'wonderful'): 1
})
self.assertEqual(a_property.get_vectorized_data(), vectorized_target)
def test_extract(self):
a_property = programfiles.Property(self._txt, "Property 1")
data = programfiles.PropertyData()
data.extract(self._pos_tags, self._noun_categories, a_property)
a_property.set_data(data)
self.assertEqual(a_property.get_data_content(),
set([('apartment', 'wonderful')]))
data = programfiles.PropertyData()
data.extract(self._pos_tags,
self._noun_categories,
a_property,
add_extra_nouns=True)
a_property.set_data(data)
self.assertEqual(
a_property.get_data_content(),
set([('apartment', None), ('apartment', 'wonderful')]))
def test_add_properties(self):
all_properties = programfiles.Properties()
a_property = programfiles.Property("What a nice property",
"Property 1")
all_properties.add_property(a_property)
self.assertEqual(all_properties.get_size(), 1)
def test_set_get_categories(self):
a_property = programfiles.Property(self._txt, "Property 1")
a_property.set_categories(self._noun_categories)
self.assertEqual(a_property.get_categories(), self._noun_categories)
def test_new_property(self):
a_property = programfiles.Property(self._txt, "Property 1")
self.assertEqual(a_property.get_name(), "Property 1")
self.assertEqual(a_property.get_p_description(), self._txt)