Example #1
0
 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]))
Example #2
0
 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]))