def test_rows_builder(self) -> None: builder = ExampleStore.builder() builder.append_row(a="a", b=1, c=True) builder.append_row(a="b", b=2, c=False) builder.append_row(a="c", b=3, c=True) test_store = builder.build() assert_that(test_store.a.tolist(), equal_to(["a", "b", "c"])) assert_that(test_store.b.tolist(), equal_to([1, 2, 3])) assert_that(test_store.c.tolist(), equal_to([True, False, True]))
def test_columns_builder(self) -> None: builder = ExampleStore.builder() builder["a"] = ["a", "b", "c"] builder.append_column("b", [1, 2, 3]).append_column(ExampleStore.c, [True, False, True]) test_store = builder.build() assert_that(test_store.a.tolist(), equal_to(["a", "b", "c"])) assert_that(test_store.b.tolist(), equal_to([1, 2, 3])) assert_that(test_store.c.tolist(), equal_to([True, False, True]))