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)