def test_conversions(self):
uut = CountVector("varname",
[lambda cursor, stack: cursor and stack],
[2])
uut.count_reference(True, True)
self.assertEqual(str(uut), "[2]")
self.assertEqual(list(uut), [2])
def test_abs(self):
uut = CountVector("varname",
[lambda x: True, lambda x: x])
self.assertEqual(abs(uut), 0)
uut.count_reference(True)
self.assertEqual(abs(uut), sqrt(2))
uut.count_reference(False)
self.assertEqual(abs(uut), sqrt(5))
def test_counting(self):
uut = CountVector("varname", [lambda cursor, stack: cursor and stack])
self.assertEqual(uut.count_vector, [0])
uut.count_reference(True, True)
self.assertEqual(uut.count_vector, [1])
uut.count_reference(True, False)
self.assertEqual(uut.count_vector, [1])
self.assertEqual(uut.unweighted, [1])
def test_cloning(self):
uut = CountVector("varname",
[lambda cursor, stack: cursor and stack],
[2])
uut.count_reference(True, True)
clone = uut.create_null_vector("test")
self.assertEqual(clone.name, "test")
self.assertEqual(clone.weightings, uut.weightings)
self.assertEqual(clone.conditions, uut.conditions)
self.assertEqual(clone.count_vector, [0])
