def test_flip(self):
table = [[1, 2, 3],['A','B','C']]
ben = BenfordDouble()
flipped = ben.flip_table(table)
self.assertEqual(1, flipped[0][0])
self.assertEqual(2, flipped[1][0])
self.assertEqual(3, flipped[2][0])
self.assertEqual('A', flipped[0][1])
self.assertEqual('B', flipped[1][1])
self.assertEqual('C', flipped[2][1])
def test_flip(self):
table = [[1, 2, 3], ['A', 'B', 'C']]
ben = BenfordDouble()
flipped = ben.flip_table(table)
self.assertEqual(1, flipped[0][0])
self.assertEqual(2, flipped[1][0])
self.assertEqual(3, flipped[2][0])
self.assertEqual('A', flipped[0][1])
self.assertEqual('B', flipped[1][1])
self.assertEqual('C', flipped[2][1])
def test_split_data(self):
ben = BenfordDouble()
output = ben.split_data("Jim, $1,233.00")
self.assertEqual("Jim", output[0])
self.assertEqual("1", output[1])
output = ben.split_data("Jim,")
self.assertEqual(None, output)
output = ben.split_data("")
self.assertEqual(None, output)
lines = "Jim, $1,233.00\n\n\3\nJim\nJim, 2,300".splitlines()
table = map(ben.split_data, lines)
table = filter(lambda x: x != None, table)
self.assertEqual(2, len(table))
def test_split_data(self):
ben = BenfordDouble()
output = ben.split_data("Jim, $1,233.00")
self.assertEqual("Jim", output[0])
self.assertEqual("1", output[1])
output = ben.split_data("Jim,")
self.assertEqual(None, output)
output = ben.split_data("")
self.assertEqual(None, output)
lines = "Jim, $1,233.00\n\n\3\nJim\nJim, 2,300".splitlines()
table = map(ben.split_data, lines)
table = filter(lambda x: x != None, table)
self.assertEqual(2, len(table))
def test_double(self):
data = "Jim, 100\n\nDave, $4,400\n\n3\nJim, 200\n\nJim,\n\nDave, 200"
ben = BenfordDouble()
table = ben.get_table(data)
self.assertEqual("Digits", table[0][0])
self.assertEqual(1, table[0][1])
self.assertEqual(9, table[0][9])
self.assertEqual("Benford", table[1][0])
self.assertEqual(30.1, table[1][1])
self.assertEqual(4.6, table[1][9])
self.assertEqual("Dave", table[2][0])
self.assertEqual(50, table[2][2])
self.assertEqual("Jim", table[3][0])
self.assertEqual(50, table[3][1])
self.assertEqual(0, table[3][3])
def test_double(self):
data = "Jim, 100\n\nDave, $4,400\n\n3\nJim, 200\n\nJim,\n\nDave, 200"
ben = BenfordDouble()
table = ben.get_table(data)
self.assertEqual("Digits", table[0][0])
self.assertEqual(1, table[0][1])
self.assertEqual(9, table[0][9])
self.assertEqual("Benford", table[1][0])
self.assertEqual(30.1, table[1][1])
self.assertEqual(4.6, table[1][9])
self.assertEqual("Dave", table[2][0])
self.assertEqual(50, table[2][2])
self.assertEqual("Jim", table[3][0])
self.assertEqual(50, table[3][1])
self.assertEqual(0, table[3][3])
