def test_data_column_data_types_with_extra_columns_also_deleted(self):
bartype = random.choice([str, int, float])
excluded_columns = ["fidelities", "foo"]
columns = Data.column_data_types(
extra_column_types={"foo": bartype},
excluded_columns=excluded_columns,
)
self.assertNotIn("fidelities", columns)
self.assertNotIn("foo", columns)
for c, t in Data.COLUMN_DATA_TYPES.items():
if c not in excluded_columns:
self.assertEqual(columns[c], t)
def test_data_column_data_types_with_extra_columns(self):
bartype = random.choice([str, int, float])
columns = Data.column_data_types(extra_column_types={"foo": bartype})
for c, t in Data.COLUMN_DATA_TYPES.items():
self.assertEqual(columns[c], t)
self.assertEqual(columns["foo"], bartype)
def test_data_column_data_types_with_removed_columns(self):
columns = Data.column_data_types(excluded_columns=["fidelities"])
self.assertNotIn("fidelities", columns)
for c, t in Data.COLUMN_DATA_TYPES.items():
if c != "fidelities":
self.assertEqual(columns[c], t)
def test_data_column_data_types_default(self):
self.assertEqual(Data.column_data_types(), Data.COLUMN_DATA_TYPES)
