def test_base_reader():
base_model = BaseModel()
X, y = base_model.load_csv(
dataset_path=os.path.join(data_dir, "cartpole_st1_at.csv"),
max_rows=1000,
augm_cols=["action_command", "config_length", "config_masspole"],
)
assert X.shape[0] == 980 == y.shape[0]
assert X.shape[1] == 7
assert y.shape[1] == 4
def test_diff_names():
base_model = BaseModel()
X, y = base_model.load_csv(
dataset_path=os.path.join(data_dir, "off_names.csv"),
input_cols=["x_position", "x_velocity", "angle_position", "angle_velocity",],
output_cols=["angle_position", "angle_velocity"],
augm_cols=["command", "length", "masspole"],
max_rows=1000,
)
assert X.shape[0] == 980 == y.shape[0]
def dal():
return BaseModel()
def dal(): #数据源
return BaseModel()
