google-research

1
// Copyright 2024 The Google Research Authors.
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
// See the License for the specific language governing permissions and
13
// limitations under the License.
14

15
#include "memory.h"
16

17
#include <cassert>
18
#include <iostream>
19

20
#include "definitions.h"
21
#include "util.h"
22
#include "gtest/gtest.h"
23

24
namespace automl_zero {
25

26
TEST(MemoryTest, NumAddressesAreWithinLimits) {
27
  const size_t max_addresses = Pow2(8 * sizeof(AddressT));
28
  EXPECT_LE(kMaxScalarAddresses, max_addresses);
29
  EXPECT_LE(kMaxVectorAddresses, max_addresses);
30
  EXPECT_LE(kMaxMatrixAddresses, max_addresses);
31
}
32

33
TEST(MemoryTest, WipeSetsValuesToZero) {
34
  const AddressT kSomeAddress = 1;
35
  const IntegerT kX = 2;
36
  const IntegerT kY = 3;
37

38
  Memory<4> memory;
39
  memory.scalar_[kSomeAddress] = 2.0;
40
  memory.vector_[kSomeAddress](kX, 0) = 4.0;
41
  memory.matrix_[kSomeAddress](kX, kY) = 0.5;
42
  EXPECT_EQ(memory.scalar_[kSomeAddress], 2.0);
43
  EXPECT_EQ(memory.vector_[kSomeAddress](kX, 0), 4.0);
44
  EXPECT_EQ(memory.matrix_[kSomeAddress](kX, kY), 0.5);
45

46
  memory.Wipe();
47
  EXPECT_EQ(memory.scalar_[kSomeAddress], 0.0);
48
  EXPECT_EQ(memory.vector_[kSomeAddress](kX, 0), 0.0);
49
  EXPECT_EQ(memory.matrix_[kSomeAddress](kX, kY), 0.0);
50
}
51

52
TEST(MemoryTest, RespectsFeaturesSize) {
53
  Memory<4> memory4;
54
  EXPECT_EQ(memory4.vector_[0].size(), 4);
55
  EXPECT_EQ(memory4.matrix_[0].rows(), 4);
56
  EXPECT_EQ(memory4.matrix_[0].cols(), 4);
57

58
  Memory<8> memory8;
59
  EXPECT_EQ(memory8.vector_[0].size(), 8);
60
  EXPECT_EQ(memory8.matrix_[0].rows(), 8);
61
  EXPECT_EQ(memory8.matrix_[0].cols(), 8);
62
}
63

64
}  // namespace automl_zero
65