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')