コード例 #1
0
ファイル: test_dataframe.py プロジェクト: ampl/amplpy
 def test_no_index(self):
     if pd is None:
         self.skipTest("pandas not available")
     df = DataFrame([], ["x", "y"])
     x = [1, 2, 3]
     y = [4, 5, 6]
     df.set_column("x", x)
     df.set_column("y", y)
     with self.assertRaises(ValueError):
         df.to_dict()
     pd_df = df.to_pandas()
     self.assertEqual(list(pd_df["x"]), x)
     self.assertEqual(list(pd_df["y"]), y)
コード例 #2
0
ファイル: test_dataframe.py プロジェクト: ampl/amplpy
 def test_dict(self):
     dic = {"aa": "bb", "c": "a"}
     self.assertEqual(dic, DataFrame.from_dict(dic).to_dict())
     dic = {1: 2}
     self.assertEqual(dic, DataFrame.from_dict(dic).to_dict())
     dic = {1: 2, 3: 4}
     self.assertEqual(dic, DataFrame.from_dict(dic).to_dict())
     dic = {2.0: ("a", "b"), 3: ("1", "2")}
     self.assertEqual(dic, DataFrame.from_dict(dic).to_dict())
     dic = {(2.0, "c"): ("a", "b"), (3, "a"): ("1", "2")}
     self.assertEqual(dic, DataFrame.from_dict(dic).to_dict())
     df = DataFrame("x", "y")
     dic = {1: 12, 2: 23}
     df.set_values(dic)
     self.assertEqual(dic, df.to_dict())
     df = DataFrame("x", ["y", "z"])
     dic = {1: (12, 2), 2: (23, -1)}
     df.set_values(dic)
     self.assertEqual(dic, df.to_dict())
     df = DataFrame("x", ["y", "z"])
     df.set_values({1: [1, 2]})
     self.assertEqual({1: (1, 2)}, df.to_dict())