def test_table_of_3(self):
"""Test a table with three elements
"""
table_3 = " 2 3 5\n2 4 6 10\n3 6 9 15\n5 10 15 25"
self.assertEqual(str(primes.primes_mult_table(3)), table_3)
def test_table_of_1(self):
"""Test a table with only one element
"""
table_1 = " 2\n2 4"
self.assertEqual(str(primes.primes_mult_table(1)), table_1)
def test_table_of_2(self):
"""Test a table with only two elements
"""
table_2 = " 2 3\n2 4 6\n3 6 9"
self.assertEqual(str(primes.primes_mult_table(2)), table_2)
def test_pandas_type(self):
"""Test if we are still using DataFrame for this function.
"""
self.assertIsInstance(primes.primes_mult_table(1),
pandas.core.frame.DataFrame)
