def testFromDataFrameExecution(self):
mdf = md.DataFrame({
'angle': [0, 3, 4],
'degree': [360, 180, 360]
},
index=['circle', 'triangle', 'rectangle'])
tensor_result = self.executor.execute_tensor(from_dataframe(mdf))
tensor_expected = self.executor.execute_tensor(
mt.tensor([[0, 360], [3, 180], [4, 360]]))
np.testing.assert_equal(tensor_result, tensor_expected)
# test up-casting
mdf2 = md.DataFrame({'a': [0.1, 0.2, 0.3], 'b': [1, 2, 3]})
tensor_result2 = self.executor.execute_tensor(from_dataframe(mdf2))
np.testing.assert_equal(tensor_result2[0].dtype, np.dtype('float64'))
tensor_expected2 = self.executor.execute_tensor(
mt.tensor([[0.1, 1.0], [0.2, 2.0], [0.3, 3.0]]))
np.testing.assert_equal(tensor_result2, tensor_expected2)
raw = [[0.1, 0.2, 0.4], [0.4, 0.7, 0.3]]
mdf3 = md.DataFrame(raw, columns=list('abc'), chunk_size=2)
tensor_result3 = self.executor.execute_tensor(from_dataframe(mdf3),
concat=True)[0]
np.testing.assert_array_equal(tensor_result3, np.asarray(raw))
self.assertTrue(tensor_result3.flags['F_CONTIGUOUS'])
self.assertFalse(tensor_result3.flags['C_CONTIGUOUS'])
def testFromDataFrameExecution(self):
mdf = md.DataFrame({
'angle': [0, 3, 4],
'degree': [360, 180, 360]
},
index=['circle', 'triangle', 'rectangle'])
tensor_result = self.executor.execute_tensor(from_dataframe(mdf))
tensor_expected = self.executor.execute_tensor(
mt.tensor([[0, 360], [3, 180], [4, 360]]))
np.testing.assert_equal(tensor_result, tensor_expected)
# test up-casting
mdf2 = md.DataFrame({'a': [0.1, 0.2, 0.3], 'b': [1, 2, 3]})
tensor_result2 = self.executor.execute_tensor(from_dataframe(mdf2))
np.testing.assert_equal(tensor_result2[0].dtype, np.dtype('float64'))
tensor_expected2 = self.executor.execute_tensor(
mt.tensor([[0.1, 1.0], [0.2, 2.0], [0.3, 3.0]]))
np.testing.assert_equal(tensor_result2, tensor_expected2)
def testFromDataFrame(self):
mdf = md.DataFrame({
'a': [0, 1, 2],
'b': [3, 4, 5]
},
index=['c', 'd', 'e'],
chunk_size=2)
tensor = from_dataframe(mdf)
self.assertEqual(tensor.shape, (3, 2))
self.assertEqual(mdf.dtypes[0], tensor.dtype)
def test_from_dataframe():
mdf = md.DataFrame({
'a': [0, 1, 2],
'b': [3, 4, 5],
'c': [0.1, 0.2, 0.3]
},
index=['c', 'd', 'e'],
chunk_size=2)
tensor = from_dataframe(mdf)
assert tensor.shape == (3, 3)
assert np.float64 == tensor.dtype
def testFromDataFrame(self):
mdf = md.DataFrame(
{
'a': [0, 1, 2],
'b': [3, 4, 5],
'c': [0.1, 0.2, 0.3]
},
index=['c', 'd', 'e'],
chunk_size=2)
tensor = from_dataframe(mdf)
self.assertEqual(tensor.shape, (3, 3))
self.assertEqual(np.float64, tensor.dtype)
