def setUp(self):
"""Setup."""
super().setUp()
self.dummy_prop_1 = IntegralProperty("Dummy 1", [])
self.dummy_prop_2 = IntegralProperty("Dummy 2", [])
self.prop = GroupedProperty("Dummy Group")
self.prop.add_property(self.dummy_prop_1)
self.prop.add_property(self.dummy_prop_2)
def test_get_property(self):
"""Test get_property."""
dummy_prop = IntegralProperty("IntegralProperty", [])
prop = GroupedProperty("Dummy Group")
prop.add_property(dummy_prop)
self.assertEqual(prop.get_property("IntegralProperty"), dummy_prop)
self.assertEqual(prop.get_property(IntegralProperty), dummy_prop)
def test_add_property(self):
"""Test add_property."""
dummy_prop = IntegralProperty("IntegralProperty", [])
prop = GroupedProperty("Dummy Group")
prop.add_property(dummy_prop)
self.assertEqual(prop._properties, {"IntegralProperty": dummy_prop})
class TestGroupedProperty(QiskitNatureTestCase):
"""General GroupedProperty base class tests."""
def setUp(self):
"""Setup."""
super().setUp()
self.dummy_prop_1 = IntegralProperty("Dummy 1", [])
self.dummy_prop_2 = IntegralProperty("Dummy 2", [])
self.prop = GroupedProperty("Dummy Group")
self.prop.add_property(self.dummy_prop_1)
self.prop.add_property(self.dummy_prop_2)
def test_init(self):
"""Test construction."""
self.assertEqual(self.prop._properties, {
"Dummy 1": self.dummy_prop_1,
"Dummy 2": self.dummy_prop_2
})
def test_add_property(self):
"""Test add_property."""
dummy_prop = IntegralProperty("IntegralProperty", [])
prop = GroupedProperty("Dummy Group")
prop.add_property(dummy_prop)
self.assertEqual(prop._properties, {"IntegralProperty": dummy_prop})
def test_get_property(self):
"""Test get_property."""
dummy_prop = IntegralProperty("IntegralProperty", [])
prop = GroupedProperty("Dummy Group")
prop.add_property(dummy_prop)
self.assertEqual(prop.get_property("IntegralProperty"), dummy_prop)
self.assertEqual(prop.get_property(IntegralProperty), dummy_prop)
def test_iter(self):
"""Test iteration.
This method also asserts that the Iterator is indeed a Generator which supports `send`.
"""
with self.subTest("Iterator 1"):
prop_list = list(self.prop.__iter__())
self.assertEqual(prop_list, [self.dummy_prop_1, self.dummy_prop_2])
with self.subTest("Iterator 2"):
prop_list = list(iter(self.prop))
self.assertEqual(prop_list, [self.dummy_prop_1, self.dummy_prop_2])
with self.subTest("Generator"):
expected = iter([self.dummy_prop_1, self.dummy_prop_2])
iterator = iter(self.prop)
prop = None
while True:
try:
prop = iterator.send(prop)
exp = next(expected)
except StopIteration:
break
self.assertEqual(prop, exp)
prop._electronic_integrals = ["Test"]
self.assertEqual(
self.prop.get_property("Dummy 1")._electronic_integrals,
["Test"])
self.assertEqual(
self.prop.get_property("Dummy 2")._electronic_integrals,
["Test"])