def test_variant_to_origin_dislacement(self, mock):
attrs = dict(self.linear_dipole_attrs, **{
'atomcharges': {'mulliken': [-1, 2]}
})
mock.configure_mock(**attrs)
x = Moments(mock).calculate(origin=[0, 0, 0])[1]
y = Moments(mock).calculate(origin=[1, 1, 1])[1]
self.assertFalse(np.array_equal(x, y))
def test_results_storing(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
mock.atomcharges.update({'lowdin': [-0.5, 0.5]})
m = Moments(mock)
m.calculate(population='mulliken')
m.calculate(population='lowdin')
a, b = m.results['mulliken'][1], m.results['lowdin'][1]
self.assertFalse(np.array_equal(a, b))
def test_configure_mock(self):
mock = Mock(foo='bar')
self.assertEqual(mock.foo, 'bar')
mock = MagicMock(foo='bar')
self.assertEqual(mock.foo, 'bar')
kwargs = {'side_effect': KeyError, 'foo.bar.return_value': 33, 'foo': MagicMock()}
mock = Mock(**kwargs)
self.assertRaises(KeyError, mock)
self.assertEqual(mock.foo.bar(), 33)
self.assertIsInstance(mock.foo, MagicMock)
mock = Mock()
mock.configure_mock(**kwargs)
self.assertRaises(KeyError, mock)
self.assertEqual(mock.foo.bar(), 33)
self.assertIsInstance(mock.foo, MagicMock)
def test_configure_mock(self):
mock = Mock(foo="bar")
self.assertEqual(mock.foo, "bar")
mock = MagicMock(foo="bar")
self.assertEqual(mock.foo, "bar")
kwargs = {"side_effect": KeyError, "foo.bar.return_value": 33, "foo": MagicMock()}
mock = Mock(**kwargs)
self.assertRaises(KeyError, mock)
self.assertEqual(mock.foo.bar(), 33)
self.assertIsInstance(mock.foo, MagicMock)
mock = Mock()
mock.configure_mock(**kwargs)
self.assertRaises(KeyError, mock)
self.assertEqual(mock.foo.bar(), 33)
self.assertIsInstance(mock.foo, MagicMock)
def parse(self, element: Element,
parserContext: ParserContext) -> BeanDefinitionInterface:
source = parserContext.extractSource(element)
defaultServletName: str = element.getattr("default-servlet-name", None)
# RootBeanDefinition use mock
RootBeanDefinition = mock.MagicMock()
mock.configure_mock(name="RootBeanDefinition")
defaultServletHandlerDef = RootBeanDefinition()
defaultServletHandlerDef.setSource(source)
defaultServletHandlerDef.setRole(
BeanDefinitionInterface.ROLE_INFRASTRUCTURE)
if defaultServletName is not None:
defaultServletHandlerDef.getPropertyValues().add(
"defaultServletName", defaultServletName)
defaultServletHandlerName: str = (parserContext.getReaderContext(
).generateBeanName(defaultServletHandlerDef))
parserContext.getRegistry().registerBeanDefinition(
defaultServletHandlerName, defaultServletHandlerDef)
parserContext.registerComponent(
BeanComponentDefinition(defaultServletHandlerDef,
defaultServletHandlerName))
urlMap = dict()
urlMap["/**"] = defaultServletHandlerName
handlerMappingDef = RootBeanDefinition()
handlerMappingDef.setSource(source)
handlerMappingDef.setRole(BeanDefinitionInterface.ROLE_INFRASTRUCTURE)
handlerMappingDef.getPropertyValues().add("urlMap", urlMap)
handlerMappingBeanName: str = (parserContext.getReaderContext().
generateBeanName(handlerMappingDef))
parserContext.getRegistry().registerBeanDefinition(
handlerMappingBeanName, handlerMappingDef)
parserContext.registerComponent(
BeanComponentDefinition(handlerMappingDef, handlerMappingBeanName))
# Ensure BeanNameUrlHandlerMapping (SPR-8289) and default HandlerAdapters are not "turned off"
registerDefaultComponents(parserContext, source)
def test_general(self):
mock = unittest.mock.MagicMock()
mock.configure_mock(**{'stop_music.return_value': None})
game = Game.from_directory(['test_levels'], [], sound_unit=mock)
level = Level.from_file('test_levels/test_level1.txt')
self.assertTrue(self.level_equals(level, game.current_level))
game.update()
level.go_next_state()
self.assertTrue(self.level_equals(level, game.current_level))
game.current_level.finished = True
game.update()
self.assertTrue(self.level_equals(level, game.current_level))
game.current_level.hp = 0
game.current_level.finished = False
game.update()
self.assertTrue(self.level_equals(level, game.current_level))
game.next_level()
level = Level.from_file('test_levels/test_level2.txt')
self.assertTrue(self.level_equals(level, game.current_level))
game.next_level()
self.assertEqual(game.over, 1)
self.assertIsNone(game.current_level)
def test_user_provided_origin(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
x = Moments(mock).calculate(origin=[1, 1, 1])
assert_almost_equal(x[0], [1, 1, 1])
def test_origin_at_center_of_mass(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
mock.atommasses = np.ones(mock.atomcoords.shape[1])
x = Moments(mock).calculate(origin='mass')[0]
assert_equal(x, [0, 0, 0])
def test_origin_at_center_of_nuclear_charge(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
x = Moments(mock).calculate()[0]
assert_equal(x, [0, 0, 0])
def test_invariant_to_origin_dislacement(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
x = Moments(mock).calculate(origin=[0, 0, 0])[1]
y = Moments(mock).calculate(origin=[1, 1, 1])[1]
assert_equal(x, y)
def test_dipole_moment(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
x = Moments(mock).calculate()[1]
assert_almost_equal(x / 4.80320425, [2, 0, 0])
def test_not_providing_masses(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
with self.assertRaisesRegex(ValueError, 'masses'):
Moments(mock).calculate(origin='mass')
def test_user_provided_masses(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
x = Moments(mock).calculate(masses=[1, 3], origin='mass')
assert_almost_equal(x[0], [0.5, 0, 0])
def test_not_providing_masses(self, mock):
mock.configure_mock(**self.linear_dipole_attrs)
# Replace with the refex version when Python2 is dropped.
# with self.assertRaisesRegex(ValueError, 'masses'):
with self.assertRaises(ValueError):
Moments(mock).calculate(origin='mass')
