def expected():
return {
(1, ): Sum(contents=np.array((4, ))),
(2, ): Sum(contents=np.array((6.2, ))),
(3, ): Sum(contents=np.array((2, ))),
(4, ): Sum(contents=np.array((2, ))),
}
def obj2():
ret = Summarizer(Summary=Sum)
ret._results.update({
(2, ): Sum(contents=np.array((3.2, ))),
(4, ): Sum(contents=np.array((2, ))),
})
return ret
def test_add_int_to_float_val(self):
obj1 = Sum(contents = [np.array((10, 20))])
obj2 = Sum(contents = [np.array((30, 40.2))])
obj3 = obj1 + obj2
np.testing.assert_equal([np.array([40, 60.2])], obj3.contents)
self.assertIsNot(obj1, obj3)
self.assertIsNot(obj1.contents, obj3.contents)
self.assertIsNot(obj2, obj3)
self.assertIsNot(obj2.contents, obj3.contents)
def test_add_key(obj):
obj.add_key('A')
expected = {'A': Sum(contents=np.array((0, )))}
assert expected == obj.results()
obj.add_key('B')
obj.add_key('C')
expected = {
'A': Sum(contents=np.array((0, ))),
'B': Sum(contents=np.array((0, ))),
'C': Sum(contents=np.array((0, ))),
}
assert expected == obj.results()
def test_add_key(self):
obj = Summarizer(Summary=Sum)
obj.add_key('A')
expected = {'A': Sum(contents=np.array((0, )))}
self.assertEqual(expected, obj.results())
obj.add_key('B')
obj.add_key('C')
expected = {
'A': Sum(contents=np.array((0, ))),
'B': Sum(contents=np.array((0, ))),
'C': Sum(contents=np.array((0, ))),
}
self.assertEqual(expected, obj.results())
def test_copy(self):
obj1 = Sum(contents = [np.array((10, 20))])
copy1 = copy.copy(obj1)
self.assertEqual(obj1, copy1)
self.assertIsNot(obj1, copy1)
self.assertIsNot(obj1.contents, copy1.contents)
self.assertIsNot(obj1.contents[0], copy1.contents[0])
def test_add(obj):
obj.add('A', (12, ))
expected = {'A': Sum(contents=np.array((12, )))}
assert expected == obj.results()
obj.add('A', (23, ))
expected = {'A': Sum(contents=np.array((35, )))}
assert expected == obj.results()
obj.add('A', (10, ), weight=2)
expected = {'A': Sum(contents=np.array((55, )))}
assert expected == obj.results()
obj.add('B', (20, ), weight=3.2)
expected = {
'A': Sum(contents=np.array((55, ))),
'B': Sum(contents=np.array((64.0, ))),
}
assert expected == obj.results()
def test_add(self):
obj = Summarizer(Summary=Sum)
obj.add('A', (12, ))
expected = {'A': Sum(contents=np.array((12, )))}
self.assertEqual(expected, obj.results())
obj.add('A', (23, ))
expected = {'A': Sum(contents=np.array((35, )))}
self.assertEqual(expected, obj.results())
obj.add('A', (10, ), weight=2)
expected = {'A': Sum(contents=np.array((55, )))}
self.assertEqual(expected, obj.results())
obj.add('B', (20, ), weight=3.2)
expected = {
'A': Sum(contents=np.array((55, ))),
'B': Sum(contents=np.array((64.0, ))),
}
self.assertEqual(expected, obj.results())
return
def setUp(self):
self.obj1 = Summarizer(Summary=Sum)
self.obj2 = Summarizer(Summary=Sum)
self.obj1._results.update({
(1, ): Sum(contents=np.array((4, ))),
(2, ): Sum(contents=np.array((3, ))),
(3, ): Sum(contents=np.array((2, ))),
})
self.obj2._results.update({
(2, ): Sum(contents=np.array((3.2, ))),
(4, ): Sum(contents=np.array((2, ))),
})
self.expected = {
(1, ): Sum(contents=np.array((4, ))),
(2, ): Sum(contents=np.array((6.2, ))),
(3, ): Sum(contents=np.array((2, ))),
(4, ): Sum(contents=np.array((2, ))),
}
def test_init_val(self):
obj = Sum(val = (10, 20))
np.testing.assert_equal([np.array([10, 20])], obj.contents)
def test_radd(self):
obj1 = Sum(contents = [np.array((10, 20))])
self.assertIsNot(obj1, sum([obj1])) # will call 0 + obj1
self.assertEqual(obj1, sum([obj1]))
def test_repr(self):
obj = Sum()
repr(obj)
def test_init_contents_not_same_object(self):
contents = [np.array([1, 3])]
obj = Sum(contents = contents)
np.testing.assert_equal(contents, obj.contents)
self.assertIsNot(contents, obj.contents)
self.assertIsNot(contents[0], obj.contents[0])
def test_init_contents(self):
obj = Sum(contents = [np.array((1, 3))])
np.testing.assert_equal([np.array([1, 3])], obj.contents)
def test_init_weight(self):
obj = Sum(val = (10, 20), weight = 2)
np.testing.assert_equal([np.array([20, 40])], obj.contents)
def test_radd_raise(self):
obj1 = Sum(contents = [np.array((10, 20))])
self.assertRaises(TypeError, obj1.__radd__, 1 ) # 1 + obj1
def test_init(self):
obj = Sum()
np.testing.assert_equal([np.array([0])], obj.contents)
