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
Example #3
0
 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()
Example #5
0
    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())
Example #6
0
 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()
Example #8
0
    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
Example #9
0
    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, ))),
        }
Example #10
0
 def test_init_val(self):
     obj = Sum(val = (10, 20))
     np.testing.assert_equal([np.array([10, 20])], obj.contents)
Example #11
0
 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]))
Example #12
0
 def test_repr(self):
     obj = Sum()
     repr(obj)
Example #13
0
 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])
Example #14
0
 def test_init_contents(self):
     obj = Sum(contents = [np.array((1, 3))])
     np.testing.assert_equal([np.array([1, 3])], obj.contents)
Example #15
0
 def test_init_weight(self):
     obj = Sum(val = (10, 20), weight = 2)
     np.testing.assert_equal([np.array([20, 40])], obj.contents)
Example #16
0
 def test_radd_raise(self):
     obj1 = Sum(contents = [np.array((10, 20))])
     self.assertRaises(TypeError, obj1.__radd__, 1 ) # 1 + obj1
Example #17
0
 def test_init(self):
     obj = Sum()
     np.testing.assert_equal([np.array([0])], obj.contents)