import unittest

from cbpred.latest_predict import OUTPUT_COLUMNS


class LatestPredictTests(unittest.TestCase):
    def test_output_columns_are_stable(self):
        self.assertEqual(
            OUTPUT_COLUMNS,
            (
                "rank",
                "symbol",
                "latest_bar_start_utc",
                "close",
                "quote_volume_usdt_8h",
                "atr_rank_200_pct",
                "drawdown_point_pct",
                "drawdown_cap_80pct_pct",
                "price_level_reached_80pct",
                "price_level_cap_80pct",
            ),
        )


if __name__ == "__main__":
    unittest.main()